Files changed (2) hide show
  1. app.py +1 -1
  2. requirements.txt +1 -1
app.py CHANGED
@@ -25,7 +25,7 @@ category_dict = {
25
  }
26
 
27
 
28
- @spaces.GPU()
29
  def yolov10_inference(image, model_id, image_size, conf_threshold, iou_threshold):
30
  model = YOLO(f"{model_id}.pt")
31
  results = model(source=image, imgsz=image_size, iou=iou_threshold, conf=conf_threshold, verbose=False)[0]
 
25
  }
26
 
27
 
28
+ @spaces.GPU(duration=200)
29
  def yolov10_inference(image, model_id, image_size, conf_threshold, iou_threshold):
30
  model = YOLO(f"{model_id}.pt")
31
  results = model(source=image, imgsz=image_size, iou=iou_threshold, conf=conf_threshold, verbose=False)[0]
requirements.txt CHANGED
@@ -1,4 +1,4 @@
1
- supervision==0.21.0
2
  ultralytics
3
  spaces
4
  gradio
 
1
+ supervision
2
  ultralytics
3
  spaces
4
  gradio