Ensure demo.launch() in __main__ for ZeroGPU @spaces.GPU function detection

app.py

@@ -14,7 +14,7 @@ from media_loader import load_media_views
 from zipsplat import ZipSplat, viz
 
 # ----------------------------------------------------------------------------
-# Output Directory & FastAPI Application
+# Output Directory & Storage
 # ----------------------------------------------------------------------------
 
 OUT_DIR = Path("outputs").resolve()
@@ -32,21 +32,6 @@ def get_viewer_template() -> str:
     return ""
 
 
-app = FastAPI()
-
-
-@app.get("/api/ply/{file_id}")
-async def get_ply(file_id: str):
-    file_path = OUT_DIR / f"{file_id}.ply"
-    if file_path.exists():
-        return FileResponse(
-            str(file_path),
-            media_type="application/octet-stream",
-            headers={"Access-Control-Allow-Origin": "*"},
-        )
-    return HTMLResponse("PLY file not found", status_code=404)
-
-
 def normalize_uploaded_paths(files: Any) -> List[Path]:
     """Normalize Gradio return types (None, str, Path, list, dict, file objects) to a list of Path objects."""
     if files is None:
@@ -215,9 +200,22 @@ with gr.Blocks(title="ZipSplatPlus: 3D Gaussian Splatting with Interactive Viewe
         outputs=[video_out, file_out, viewer_out, status_out],
     )
 
-# Mount Gradio demo onto FastAPI app
+# FastAPI app for serving raw PLY files
+app = FastAPI()
+
+@app.get("/api/ply/{file_id}")
+async def get_ply(file_id: str):
+    file_path = OUT_DIR / f"{file_id}.ply"
+    if file_path.exists():
+        return FileResponse(
+            str(file_path),
+            media_type="application/octet-stream",
+            headers={"Access-Control-Allow-Origin": "*"},
+        )
+    return HTMLResponse("PLY file not found", status_code=404)
+
+# Mount Gradio app onto FastAPI
 app = gr.mount_gradio_app(app, demo, path="/", allowed_paths=[str(OUT_DIR), tempfile.gettempdir()])
 
 if __name__ == "__main__":
-    import uvicorn
-    uvicorn.run("app:app", host="0.0.0.0", port=7860)
+    demo.launch()