kadirnar commited on
Commit
e78a6e2
·
verified ·
1 Parent(s): 2a05c12

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -45,7 +45,7 @@ def attempt_download_from_hub(repo_id, hf_token=None):
45
  return None
46
 
47
 
48
- @spaces.GPU()
49
  def LeYOLO_inference(image, model_id, image_size, conf_threshold, iou_threshold):
50
  MODEL_PATH = attempt_download_from_hub(model_id)
51
  model = model = YOLO(MODEL_PATH)
 
45
  return None
46
 
47
 
48
+ @spaces.GPU(duration=10)
49
  def LeYOLO_inference(image, model_id, image_size, conf_threshold, iou_threshold):
50
  MODEL_PATH = attempt_download_from_hub(model_id)
51
  model = model = YOLO(MODEL_PATH)