🚀

app.py

@@ -7,6 +7,7 @@ import io
 import cv2
 from typing import Dict, Any, List, Optional
 from transformers import Sam3Model, Sam3Processor
+import spaces
 
 class SAM3Handler:
     """SAM3 handler for both UI and API access"""
@@ -24,6 +25,7 @@ class SAM3Handler:
         self.processor = Sam3Processor.from_pretrained("facebook/sam3")
         print("SAM3 model loaded successfully")
 
+    @spaces.GPU
     def predict(self, image, text_prompt, confidence_threshold=0.5):
         """
         Main prediction function for both UI and API

requirements.txt

@@ -9,4 +9,5 @@ accelerate
 opencv-python
 huggingface-hub
 tokenizers
-gradio
+gradio
+spaces