linhcuem commited on
Commit
6e8c52d
1 Parent(s): 8060d72

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -23,7 +23,7 @@ image_path = [['test_images/2a998cfb0901db5f8210.jpg','linhcuem/cham_diem_yolov8
23
 
24
  # Load YOLO model
25
  model = YOLO('linhcuem/cham_diem_yolov8')
26
- model = YOLO('linhcuem/cham_diem_yolov8_ver20')
27
 
28
  ###################################################
29
  def yolov8_img_inference(
@@ -82,7 +82,7 @@ def yolov8_img_inference(
82
  inputs_image = [
83
  gr.inputs.Image(type="filepath", label="Input Image"),
84
  gr.inputs.Dropdown(["linhcuem/linhcuem/cham_diem_yolov8"],
85
- default="linhcuem/cham_diem_yolov8_ver20", label="Model"),
86
  gr.inputs.Slider(minimum=320, maximum=1280, default=640, step=32, label="Image Size"),
87
  gr.inputs.Slider(minimum=0.0, maximum=1.0, default=0.25, step=0.05, label="Confidence Threshold"),
88
  gr.inputs.Slider(minimum=0.0, maximum=1.0, default=0.45, step=0.05, label="IOU Threshold"),
 
23
 
24
  # Load YOLO model
25
  model = YOLO('linhcuem/cham_diem_yolov8')
26
+ # model = YOLO('linhcuem/cham_diem_yolov8_ver20')
27
 
28
  ###################################################
29
  def yolov8_img_inference(
 
82
  inputs_image = [
83
  gr.inputs.Image(type="filepath", label="Input Image"),
84
  gr.inputs.Dropdown(["linhcuem/linhcuem/cham_diem_yolov8"],
85
+ default="linhcuem/cham_diem_yolov8", label="Model"),
86
  gr.inputs.Slider(minimum=320, maximum=1280, default=640, step=32, label="Image Size"),
87
  gr.inputs.Slider(minimum=0.0, maximum=1.0, default=0.25, step=0.05, label="Confidence Threshold"),
88
  gr.inputs.Slider(minimum=0.0, maximum=1.0, default=0.45, step=0.05, label="IOU Threshold"),