Spaces:
Sleeping
Sleeping
Update app.py
Browse files
app.py
CHANGED
@@ -34,7 +34,7 @@ image_path = [['test_images/2a998cfb0901db5f8210.jpg','cham_diem_yolov8', 640, 0
|
|
34 |
# model_ids = ['linhcuem/checker_TB_yolov8_ver1', 'linhcuem/cham_diem_yolov8', 'linhcuem/chamdiemgianhang_yolov8_300epochs', 'linhcuem/cham_diem_yolov8_ver20', 'linhcuem/chamdiemgianhang_yolov8_ver21']
|
35 |
# current_model_id = model_ids[-1]
|
36 |
# model = YOLO(current_model_id)
|
37 |
-
|
38 |
|
39 |
###################################################
|
40 |
def yolov8_img_inference(
|
@@ -44,7 +44,7 @@ def yolov8_img_inference(
|
|
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
|
49 |
model.iou = iou_threshold
|
50 |
# model.overrides['conf'] = conf_threshold
|
|
|
34 |
# model_ids = ['linhcuem/checker_TB_yolov8_ver1', 'linhcuem/cham_diem_yolov8', 'linhcuem/chamdiemgianhang_yolov8_300epochs', 'linhcuem/cham_diem_yolov8_ver20', 'linhcuem/chamdiemgianhang_yolov8_ver21']
|
35 |
# current_model_id = model_ids[-1]
|
36 |
# model = YOLO(current_model_id)
|
37 |
+
model = YOLO(model_path)
|
38 |
|
39 |
###################################################
|
40 |
def yolov8_img_inference(
|
|
|
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
|
49 |
model.iou = iou_threshold
|
50 |
# model.overrides['conf'] = conf_threshold
|