Spaces:
Sleeping
Sleeping
Update app.py
Browse files
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",
|
|
|
140 |
default="linhcuem/checker_TB_yolov8_ver1", label="Model"),
|
141 |
|
142 |
-
gr.Slider(maximum=1280, step=32,
|
143 |
-
gr.Slider(maximum=1.0, step=0.05,
|
144 |
-
gr.Slider(maximum=1.0, step=0.05,
|
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,
|