Update app.py

app.py

@@ -8,6 +8,25 @@ from sorawm.core import SoraWM
 
 
 def main():
+    st.set_page_config(page_title="Sora Watermark Cleaner", page_icon="🎬", layout="centered")
+    st.title("🎬 Sora Watermark Cleaner")
+    st.markdown("Remove watermarks from Sora-generated videos with ease")
+
+    # Problem line - wrap with exception handler
+    try:
+        if "sora_wm" not in st.session_state:
+            with st.spinner("Loading AI models..."):
+                import sorawm.core
+                st.session_state.sora_wm = sorawm.core.SoraWM()
+        st.success("Model loaded.")
+    except Exception as e:
+        st.error(f"Model load failed: {e}")
+        import traceback
+        st.text(traceback.format_exc())
+        return
+
+
+    
     st.set_page_config(
         page_title="Sora Watermark Cleaner", page_icon="🎬", layout="centered"
     )