linhcuem commited on
Commit
41f369f
·
verified ·
1 Parent(s): 023264e

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +6 -5
app.py CHANGED
@@ -135,13 +135,14 @@ title = "Detect Thiên Việt productions"
135
  interface_image = gr.Interface(
136
  fn=yolov8_img_inference,
137
  inputs=[
138
- gr.Image(type='pil'),
139
- gr.Dropdown(["linhcuem/checker_TB_yolov8_ver1","linhcuem/cham_diem_yolov8","linhcuem/chamdiemgianhang_yolov8_300epochs", "linhcuem/chamdiemgianhang_yolov8_ver21", "linhcuem/cham_diem_yolov8_ver20" ],
 
140
  default="linhcuem/checker_TB_yolov8_ver1", label="Model"),
141
 
142
- gr.Slider(maximum=1280, step=32, value=640),
143
- gr.Slider(maximum=1.0, step=0.05, value=0.25),
144
- gr.Slider(maximum=1.0, step=0.05, value=0.45),
145
  ],
146
  outputs=[gr.Image(type="pil"),gr.Textbox(show_label=False)],
147
  title=title,
 
135
  interface_image = gr.Interface(
136
  fn=yolov8_img_inference,
137
  inputs=[
138
+ gr.inputs.Image(type='pil'),
139
+ gr.inputs.Dropdown(["linhcuem/checker_TB_yolov8_ver1", "linhcuem/cham_diem_yolov8", "linhcuem/chamdiemgianhang_yolov8_300epochs",
140
+ "linhcuem/chamdiemgianhang_yolov8_ver21", "linhcuem/cham_diem_yolov8_ver20"],
141
  default="linhcuem/checker_TB_yolov8_ver1", label="Model"),
142
 
143
+ gr.inputs.Slider(minimum=320, maximum=1280, default=640, step=32, label="Image Size"),
144
+ gr.inputs.Slider(minimum=0.0, maximum=1.0, default=0.25, step=0.05, label="Confidence Threshold"),
145
+ gr.inputs.Slider(minimum=0.0, maximum=1.0, default=0.45, step=0.05, label="IOU Threshold"),
146
  ],
147
  outputs=[gr.Image(type="pil"),gr.Textbox(show_label=False)],
148
  title=title,