Update app.py

app.py

@@ -2,13 +2,7 @@
 # This file is required and must be named app.py for HF Spaces
 
 # Import your main script
-from ocs4dev import create_demo  # Make sure this function exists and returns a gr.Blocks object
+from ocs4dev import main
 
 if __name__ == "__main__":
-    # Create and launch the demo
-    demo = create_demo()
-    demo.launch()
-
-# For HF Spaces, also define at module level
-demo = create_demo()
-demo.launch()
+    main()