Spaces:
Running
Running
Update app.py
Browse files
app.py
CHANGED
@@ -37,20 +37,20 @@ def yolov8_img_inference(
|
|
37 |
|
38 |
return render
|
39 |
|
40 |
-
|
41 |
gr.inputs.Image(type="filepath", label="Input Image"),
|
42 |
-
gr.inputs.Dropdown(["linhcuem/chamdiem_yolov8_ver10"],
|
43 |
-
|
44 |
gr.inputs.Slider(minimum=320, maximum=1280, default=640, step=32, label="Image Size"),
|
45 |
-
gr.inputs.Slider(minimum=0.0, maximum=1.0, default=
|
46 |
-
gr.inputs.Slider(minimum=0.0, maximum=1.0, default=0.45, step=0.05, label="IOU Threshold"),
|
47 |
]
|
48 |
|
49 |
-
|
50 |
interface_image = gr.Interface(
|
51 |
fn=yolov8_img_inference,
|
52 |
-
inputs=
|
53 |
-
outputs=
|
54 |
title=model_heading,
|
55 |
description=description,
|
56 |
examples=image_path,
|
|
|
37 |
|
38 |
return render
|
39 |
|
40 |
+
inputs_image = [
|
41 |
gr.inputs.Image(type="filepath", label="Input Image"),
|
42 |
+
gr.inputs.Dropdown(["linhcuem/chamdiem_yolov8_ver10"],
|
43 |
+
default="linhcuem/chamdiem_yolov8_ver10", label="Model"),
|
44 |
gr.inputs.Slider(minimum=320, maximum=1280, default=640, step=32, label="Image Size"),
|
45 |
+
gr.inputs.Slider(minimum=0.0, maximum=1.0, default=0.25, step=0.05, label="Confidence Threshold"),
|
46 |
+
gr.inputs.Slider(minimum=0.0, maximum=1.0, default=0.45, step=0.05, label="IOU Threshold"),
|
47 |
]
|
48 |
|
49 |
+
outputs_image =gr.outputs.Image(type="filepath", label="Output Image")
|
50 |
interface_image = gr.Interface(
|
51 |
fn=yolov8_img_inference,
|
52 |
+
inputs=inputs_image,
|
53 |
+
outputs=outputs_image,
|
54 |
title=model_heading,
|
55 |
description=description,
|
56 |
examples=image_path,
|