Update app.py

app.py

@@ -14,6 +14,7 @@ model_mm_grounding_id = "rziga/mm_grounding_dino_tiny_o365v1_goldg"
 model_llmdet_name = extract_model_short_name(model_llmdet_id)
 model_mm_grounding_name = extract_model_short_name(model_mm_grounding_id)
 
+@spaces.GPU
 def detect_llmdet(image: Image.Image, prompts: list, threshold: float):
     t0 = time.perf_counter()
     model_id = model_llmdet_id
@@ -42,6 +43,7 @@ def detect_llmdet(image: Image.Image, prompts: list, threshold: float):
     raw_text = "\n".join(raw_results) if raw_results else "No detections"
     return annotations, raw_text, time_taken
 
+@spaces.GPU
 def detect_mm_grounding(image: Image.Image, prompts: list, threshold: float):
     t0 = time.perf_counter()
     model_id = model_mm_grounding_id
@@ -70,7 +72,7 @@ def detect_mm_grounding(image: Image.Image, prompts: list, threshold: float):
     raw_text = "\n".join(raw_results) if raw_results else "No detections"
     return annotations, raw_text, time_taken
 
-@spaces.GPU
+
 def run_detection(image, prompts_str, threshold):
     if image is None:
         return (None, []), "No detections", "", (None, []), "No detections", ""