linhcuem commited on
Commit
ba0ecd2
·
1 Parent(s): 0fe6d0c

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -4
app.py CHANGED
@@ -26,8 +26,8 @@ image_path = [['test_images/2a998cfb0901db5f8210.jpg','linhcuem/cham_diem_yolov8
26
  ['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]]
27
 
28
  # Load YOLO model
29
- # model = YOLO('linhcuem/cham_diem_yolov8')
30
- model = YOLO('linhcuem/cham_diem_yolov8_ver20')
31
 
32
  ###################################################
33
  def yolov8_img_inference(
@@ -128,8 +128,8 @@ interface_image = gr.Interface(
128
  fn=yolov8_img_inference,
129
  inputs=[
130
  gr.Image(type='pil'),
131
- gr.Dropdown(["linhcuem/cham_diem_yolov8_ver20"],
132
- default="linhcuem/cham_diem_yolov8_ver20"),
133
  gr.Slider(maximum=1280, step=32, value=640),
134
  gr.Slider(maximum=1.0, step=0.05, value=0.25),
135
  gr.Slider(maximum=1.0, step=0.05, value=0.45),
 
26
  ['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]]
27
 
28
  # Load YOLO model
29
+ model = YOLO('linhcuem/cham_diem_yolov8')
30
+ # model = YOLO('linhcuem/cham_diem_yolov8_ver20')
31
 
32
  ###################################################
33
  def yolov8_img_inference(
 
128
  fn=yolov8_img_inference,
129
  inputs=[
130
  gr.Image(type='pil'),
131
+ gr.Dropdown(["linhcuem/cham_diem_yolov8"],
132
+ default="linhcuem/cham_diem_yolov8"),
133
  gr.Slider(maximum=1280, step=32, value=640),
134
  gr.Slider(maximum=1.0, step=0.05, value=0.25),
135
  gr.Slider(maximum=1.0, step=0.05, value=0.45),