Hugging Face's logo

Spaces:
linhcuem
/
cham_diem_yolov8
Sleeping

App Files Community
linhcuem commited on
Commit
8e98827
·
verified ·
1 Parent(s): c61869a

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -1
app.py CHANGED
@@ -46,8 +46,9 @@ def yolov8_img_inference(
46
  image_size: gr.inputs.Slider = 640,
47
  conf_threshold: gr.inputs.Slider = 0.25,
48
  iou_threshold: gr.inputs.Slider = 0.45,
 
49
  ):
50
- model_path = download_from_hub(model_path)
51
  model = YOLO(model_path)
52
 
53
  model.conf = conf_threshold
 
46
  image_size: gr.inputs.Slider = 640,
47
  conf_threshold: gr.inputs.Slider = 0.25,
48
  iou_threshold: gr.inputs.Slider = 0.45,
49
+ token: gr.inputs.Text = None
50
  ):
51
+ model_path = download_from_hub(model_path, token)
52
  model = YOLO(model_path)
53
 
54
  model.conf = conf_threshold