fixed bugs

demo_gradio.py

@@ -2,6 +2,12 @@ import os
 
 import gradio as gr
 
+try:
+    import spaces
+    HAS_SPACES = True
+except ImportError:
+    HAS_SPACES = False
+
 from longstream.demo import BRANCH_OPTIONS, create_demo_session, load_metadata
 from longstream.demo.backend import load_frame_previews
 from longstream.demo.export import export_glb
@@ -14,7 +20,7 @@ DEFAULT_WINDOW_SIZE = 48
 DEFAULT_CHECKPOINT = os.getenv("LONGSTREAM_CHECKPOINT", "checkpoints/50_longstream.pt")
 
 
-def _run_stable_demo(
+def _run_stable_demo_impl(
     image_dir,
     uploaded_files,
     uploaded_video,
@@ -97,6 +103,12 @@ def _run_stable_demo(
     )
 
 
+if HAS_SPACES:
+    _run_stable_demo = spaces.GPU(_run_stable_demo_impl)
+else:
+    _run_stable_demo = _run_stable_demo_impl
+
+
 def _update_stable_scene(
     session_dir,
     branch_label,