fix: temporarily migrate away from ZeroGPU

app.py

@@ -1,11 +1,8 @@
 from typing import Any, Dict
 
-import spaces
-
 import cv2
 import gradio as gr
 import numpy as np
-import torch
 from gradio_image_annotation import image_annotator
 from sam2 import load_model
 from sam2.sam2_image_predictor import SAM2ImagePredictor
@@ -13,14 +10,12 @@ from sam2.sam2_image_predictor import SAM2ImagePredictor
 from src.plot_utils import export_mask
 
 
-@spaces.GPU()
+# @spaces.GPU()
 def predict(model_choice, annotations: Dict[str, Any]):
-    # device = "cuda" if torch.cuda.is_available() else "cpu"
-    torch.autocast(device_type="cuda", dtype=torch.bfloat16).__enter__()
     sam2_model = load_model(
         variant=model_choice,
         ckpt_path=f"assets/checkpoints/sam2_hiera_{model_choice}.pt",
-        device="cuda",
+        device="cpu",
     )
     predictor = SAM2ImagePredictor(sam2_model)  # type:ignore
     predictor.set_image(annotations["image"])