linhcuem commited on
Commit
9b71308
1 Parent(s): 82992b1

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -3
app.py CHANGED
@@ -27,7 +27,8 @@ image_path = [['test_images/2a998cfb0901db5f8210.jpg','linhcuem/cham_diem_yolov8
27
  ['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]]
28
 
29
  # Load YOLO model
30
- model = YOLO('linhcuem/cham_diem_yolov8')
 
31
  # model = YOLO('linhcuem/cham_diem_yolov8_ver20')
32
 
33
  ###################################################
@@ -164,8 +165,8 @@ interface_image = gr.Interface(
164
  fn=yolov8_img_inference,
165
  inputs=[
166
  gr.Image(type='pil'),
167
- gr.Dropdown(["linhcuem/cham_diem_yolov8_ver20"],
168
- default="linhcuem/cham_diem_yolov8_ver20"),
169
  gr.Slider(maximum=1280, step=32, value=640),
170
  gr.Slider(maximum=1.0, step=0.05, value=0.25),
171
  gr.Slider(maximum=1.0, step=0.05, value=0.45),
 
27
  ['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]]
28
 
29
  # Load YOLO model
30
+ # model = YOLO('linhcuem/cham_diem_yolov8')
31
+ model = YOLO('linhcuem/chamdiemgianhang_yolov8_ver1')
32
  # model = YOLO('linhcuem/cham_diem_yolov8_ver20')
33
 
34
  ###################################################
 
165
  fn=yolov8_img_inference,
166
  inputs=[
167
  gr.Image(type='pil'),
168
+ gr.Dropdown(["linhcuem/chamdiemgianhang_yolov8_ver1"],
169
+ default="linhcuem/chamdiemgianhang_yolov8_ver1"),
170
  gr.Slider(maximum=1280, step=32, value=640),
171
  gr.Slider(maximum=1.0, step=0.05, value=0.25),
172
  gr.Slider(maximum=1.0, step=0.05, value=0.45),