Spaces:
Sleeping
Sleeping
Update app.py
Browse files
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/
|
84 |
-
default="linhcuem/
|
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"),
|