Spaces:
Sleeping
Sleeping
Update app.py
Browse files
app.py
CHANGED
@@ -38,6 +38,7 @@ image_path = [['test_images/2a998cfb0901db5f8210.jpg','cham_diem_yolov8', 640, 0
|
|
38 |
# current_model_id = model_ids[-1]
|
39 |
# model = YOLO(current_model_id)
|
40 |
# model = YOLO(model_path)
|
|
|
41 |
|
42 |
###################################################
|
43 |
def yolov8_img_inference(
|
@@ -46,9 +47,9 @@ def yolov8_img_inference(
|
|
46 |
image_size: gr.inputs.Slider = 640,
|
47 |
conf_threshold: gr.inputs.Slider = 0.25,
|
48 |
iou_threshold: gr.inputs.Slider = 0.45,
|
49 |
-
|
50 |
):
|
51 |
-
|
52 |
model = YOLO(model_path)
|
53 |
|
54 |
model.conf = conf_threshold
|
|
|
38 |
# current_model_id = model_ids[-1]
|
39 |
# model = YOLO(current_model_id)
|
40 |
# model = YOLO(model_path)
|
41 |
+
model_path = download_from_hub("linhcuem/checker_TB_yolov8_ver1", "linhcuem/chamdiemgianhang_yolov8_ver21")
|
42 |
|
43 |
###################################################
|
44 |
def yolov8_img_inference(
|
|
|
47 |
image_size: gr.inputs.Slider = 640,
|
48 |
conf_threshold: gr.inputs.Slider = 0.25,
|
49 |
iou_threshold: gr.inputs.Slider = 0.45,
|
50 |
+
|
51 |
):
|
52 |
+
|
53 |
model = YOLO(model_path)
|
54 |
|
55 |
model.conf = conf_threshold
|