Update app.py

app.py

@@ -8,7 +8,7 @@ from sorawm.core import SoraWM
 
 
 def main():
-     st.set_page_config(
+    st.set_page_config(
         page_title="Sora Watermark Cleaner", page_icon="🎬", layout="centered"
     )
 
@@ -16,7 +16,9 @@ def main():
     st.markdown("Remove watermarks from Sora-generated videos with ease")
 
     # Initialize SoraWM
-    st.write("Loaded UI without model")
+    if "sora_wm" not in st.session_state:
+        with st.spinner("Loading AI models..."):
+            st.session_state.sora_wm = SoraWM()
 
     st.markdown("---")