--DELETE--

app.py

@@ -1,6 +1,7 @@
 import gradio as gr
 from ultralytics import YOLO
 import supervision as sv
+import spaces
 
     
 box_annotator = sv.BoxAnnotator()
@@ -24,7 +25,7 @@ category_dict = {
 }
 
 
-
+@spaces.GPU(duration=200)
 def yolov10_inference(image, model_id, image_size, conf_threshold, iou_threshold):
     model = YOLO(f"{model_id}.pt")
     results = model(source=image, imgsz=image_size, iou=iou_threshold, conf=conf_threshold, verbose=False)[0]

requirements.txt

@@ -1,4 +1,4 @@
-supervision==0.21.0
+supervision
 ultralytics
 spaces
 gradio