Update app.py

app.py

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