Update app.py

app.py

@@ -1,5 +1,5 @@
 import gradio as gr
-
+import spaces
 import argparse
 import cv2
 from PIL import Image
@@ -23,6 +23,7 @@ device = "cuda" if torch.cuda.is_available() else "cpu"
 processor = AutoProcessor.from_pretrained(model_id)
 model = AutoModelForZeroShotObjectDetection.from_pretrained(model_id).to(device)
 
+@spaces.GPU
 def run_grounding(input_image, grounding_caption, box_threshold, text_threshold):
     # Convert numpy array to PIL Image if needed
     if isinstance(input_image, np.ndarray):