StevenChen16 commited on
Commit
6105a60
·
verified ·
1 Parent(s): fcd27de

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -31,7 +31,7 @@ category_dict = {
31
  }
32
 
33
 
34
- @spaces.GPU(duration=1000)
35
  def yolov10_inference(image, video, model_id, image_size, conf_threshold, iou_threshold):
36
  model_path = download_models(model_id)
37
  model = YOLOv10.from_pretrained(model_path)
 
31
  }
32
 
33
 
34
+ @spaces.GPU(duration=200)
35
  def yolov10_inference(image, video, model_id, image_size, conf_threshold, iou_threshold):
36
  model_path = download_models(model_id)
37
  model = YOLOv10.from_pretrained(model_path)