Support Huggingface GPU

app.py

@@ -15,6 +15,7 @@ from datetime import datetime
 import glob
 import gc
 import time
+import spaces
 
 from pi3.utils.geometry import se3_inverse, homogenize_points, depth_edge
 from pi3.models.pi3 import Pi3
@@ -274,6 +275,7 @@ def compute_camera_faces(cone_shape: trimesh.Trimesh) -> np.ndarray:
 # -------------------------------------------------------------------------
 # 1) Core model inference
 # -------------------------------------------------------------------------
+@spaces.GPU(duration=120)
 def run_model(target_dir, model) -> dict:
     print(f"Processing images from {target_dir}")
 
@@ -415,6 +417,7 @@ def update_gallery_on_upload(input_video, input_images, interval=-1):
 # -------------------------------------------------------------------------
 # 4) Reconstruction: uses the target_dir plus any viz parameters
 # -------------------------------------------------------------------------
+@spaces.GPU(duration=120)
 def gradio_demo(
     target_dir,
     conf_thres=3.0,

requirements.txt

@@ -11,4 +11,5 @@ safetensors
 gradio
 trimesh
 matplotlib
-scipy
+scipy
+spaces