Hugging Face's logo

Spaces:
perctrix
/
YOLOv10
Running

App Files Community
Steven Chen commited on
Commit
4a5d0fb
·
verified ·
1 Parent(s): ffa2e5d

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -0
app.py CHANGED
@@ -32,6 +32,7 @@ category_dict = {
32
  }
33
 
34
 
 
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)
 
32
  }
33
 
34
 
35
+ @spaces.GPU
36
  def yolov10_inference(image, video, model_id, image_size, conf_threshold, iou_threshold):
37
  model_path = download_models(model_id)
38
  model = YOLOv10(model_path)