feat: migrate to ZeroGPU

app.py

@@ -2,6 +2,8 @@ import gradio as gr
 import numpy as np
 import cv2
 
+import spaces
+
 from PIL import Image
 
 from src.plot_utils import show_masks
@@ -18,7 +20,7 @@ choice_mapping = {
     "large": ["sam2_hiera_l.yaml", "assets/checkpoints/sam2_hiera_large.pt"],
 }
 
-
+@spaces.GPU
 def predict(model_choice: str, annotations, image):
     config_file, ckpt_path = choice_mapping[str(model_choice)]
     sam2_model = build_sam2(config_file, ckpt_path, device="cpu")

requirements.txt

@@ -1,4 +1,5 @@
+git+https://github.com/facebookresearch/segment-anything-2.git
 gradio
 gradio_image_annotation
 opencv-python
-samv2
+spaces