Update app_py.py

app_py.py

@@ -1,8 +1,14 @@
-# app.py - Entry point for Hugging Face Spaces
-# This file is required and must be named app.py for HF Spaces
-
-# Import your main script
-from ocs4dev import main
-
-if __name__ == "__main__":
-    main()
+# app.py - Entry point for Hugging Face Spaces
+# 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
+
+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()