Update app.py

app.py

@@ -4,6 +4,7 @@ import tempfile
 from ultralytics import YOLOv10
 import supervision as sv
 from huggingface_hub import hf_hub_download
+import spaces
 
 
 def download_models(model_id):
@@ -30,7 +31,7 @@ category_dict = {
     77: 'teddy bear', 78: 'hair drier', 79: 'toothbrush'
 }
 
-
+@spaces.GPU(duration=200)
 def yolov10_inference(image, video, model_id, image_size, conf_threshold, iou_threshold):
     model_path = download_models(model_id)
     model = YOLOv10(model_path)