add spaces

app.py

@@ -26,6 +26,7 @@ from datetime import (
 )
 import cv2
 import numpy as np
+import spaces
 
 def prepare_to_run_demo():
     config = load_config('configs/infer.json')
@@ -65,6 +66,7 @@ def ply2glb(ply_path, glb_path):
     cloud.export(glb_path)
     os.remove(ply_path)
 
+@spaces.GPU
 def fn(image_path, mask_path):
     device = "cuda" if torch.cuda.is_available() else "cpu"
     name_base, _ = os.path.splitext(os.path.basename(image_path))

requirements.txt

@@ -16,4 +16,5 @@ einops==0.8.0
 timm==1.0.15
 trimesh==4.5.2
 transformers==4.46.3
-matplotlib==3.9.2
+matplotlib==3.9.2
+spaces==0.42.1