fix?: necessarily use CUDA

app.py

@@ -15,11 +15,11 @@ from src.plot_utils import export_mask
 
 @spaces.GPU()
 def predict(model_choice, annotations: Dict[str, Any]):
-    device = "cuda" if torch.cuda.is_available() else "cpu"
+    # device = "cuda" if torch.cuda.is_available() else "cpu"
     sam2_model = load_model(
         variant=model_choice,
         ckpt_path=f"assets/checkpoints/sam2_hiera_{model_choice}.pt",
-        device=device,
+        device="cuda",
     )
     predictor = SAM2ImagePredictor(sam2_model)  # type:ignore
     predictor.set_image(annotations["image"])