Spaces:
Sleeping
Sleeping
Update app.py
Browse files
app.py
CHANGED
@@ -38,11 +38,11 @@ image_path = [['test_images/2a998cfb0901db5f8210.jpg','cham_diem_yolov8', 640, 0
|
|
38 |
|
39 |
###################################################
|
40 |
def yolov8_img_inference(
|
41 |
-
image,
|
42 |
-
model_path= None,
|
43 |
-
image_size= 640,
|
44 |
-
conf_threshold= 0.25,
|
45 |
-
iou_threshold = 0.45,
|
46 |
):
|
47 |
model = YOLO(model_path)
|
48 |
model.conf = conf_threshold
|
|
|
38 |
|
39 |
###################################################
|
40 |
def yolov8_img_inference(
|
41 |
+
image: gr.inputs.Image = None,
|
42 |
+
model_path: gr.inputs.Dropdown = None,
|
43 |
+
image_size: gr.inputs.Slider = 640,
|
44 |
+
conf_threshold: gr.inputs.Slider = 0.25,
|
45 |
+
iou_threshold: gr.inputs.Slider = 0.45,
|
46 |
):
|
47 |
model = YOLO(model_path)
|
48 |
model.conf = conf_threshold
|