add device

app.py

@@ -65,11 +65,12 @@ 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))
     config, accelerator = prepare_to_run_demo()
     model = load_model(config, accelerator)
+    model = model.to(device)
     image, cv2_image, mask = load_infer_data_demo(image_path, mask_path, 
         model_dtype=config['spherevit']['dtype'], device=accelerator.device)
     if torch.backends.mps.is_available():