linhcuem commited on
Commit
68128b9
1 Parent(s): 6578a56

Update app.py

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