Update app.py

app.py

@@ -34,6 +34,7 @@ def yolov9_inference(model_id, image_size, conf_threshold, iou_threshold, img_pa
 
     return output_image, output_video, output_path
 
+@spaces.GPU(duration=60)
 def inference(model_id, image_size, conf_threshold, iou_threshold, img_path=None, vid_path=None):
     global should_continue
     should_continue = True