Update app.py

app.py

@@ -13,6 +13,7 @@ import base64
 import cv2
 from typing import List, Tuple, Optional
 import sys
+import spaces
 
 
 def add_sam2_to_path():
@@ -230,6 +231,8 @@ def visualize_masks_on_image(
     blended = cv2.addWeighted(image_np, 1 - alpha, color_mask, alpha, 0)
     return PILImage.fromarray(blended)
 
+@torch.no_grad()
+@spaces.GPU(duration=60)
 def run_pipeline(image: PILImage.Image, prompt: str):
     if not model or not processor:
         return "Models not loaded. Please check logs.", None