StevenChen16 commited on
Commit
3983c11
1 Parent(s): 14d3c09

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
  77: 'teddy bear', 78: 'hair drier', 79: 'toothbrush'
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(model_path)
 
31
  77: 'teddy bear', 78: 'hair drier', 79: 'toothbrush'
32
  }
33
 
34
+ @spaces.GPU(duration=120)
35
  def yolov10_inference(image, video, model_id, image_size, conf_threshold, iou_threshold):
36
  model_path = download_models(model_id)
37
  model = YOLOv10(model_path)