Spaces:
Sleeping
Sleeping
Update app.py
Browse files
app.py
CHANGED
@@ -33,7 +33,7 @@ image_path = [['test_images/2a998cfb0901db5f8210.jpg','cham_diem_yolov8', 640, 0
|
|
33 |
# model = YOLO('linhcuem/chamdiemgianhang_yolov8_ver21')
|
34 |
# model = YOLO('linhcuem/cham_diem_yolov8_ver20')
|
35 |
# model_ids = ['linhcuem/checker_TB_yolov8_ver1', 'linhcuem/cham_diem_yolov8', 'linhcuem/chamdiemgianhang_yolov8_300epochs', 'linhcuem/cham_diem_yolov8_ver20', 'linhcuem/chamdiemgianhang_yolov8_ver21']
|
36 |
-
|
37 |
# model = YOLO(model_path)
|
38 |
# current_model_id = model_ids[-1]
|
39 |
# model = YOLO(current_model_id)
|
@@ -47,6 +47,7 @@ def yolov8_img_inference(
|
|
47 |
conf_threshold: gr.inputs.Slider = 0.25,
|
48 |
iou_threshold: gr.inputs.Slider = 0.45,
|
49 |
):
|
|
|
50 |
model = YOLO(model_path)
|
51 |
|
52 |
model.conf = conf_threshold
|
|
|
33 |
# model = YOLO('linhcuem/chamdiemgianhang_yolov8_ver21')
|
34 |
# model = YOLO('linhcuem/cham_diem_yolov8_ver20')
|
35 |
# model_ids = ['linhcuem/checker_TB_yolov8_ver1', 'linhcuem/cham_diem_yolov8', 'linhcuem/chamdiemgianhang_yolov8_300epochs', 'linhcuem/cham_diem_yolov8_ver20', 'linhcuem/chamdiemgianhang_yolov8_ver21']
|
36 |
+
|
37 |
# model = YOLO(model_path)
|
38 |
# current_model_id = model_ids[-1]
|
39 |
# model = YOLO(current_model_id)
|
|
|
47 |
conf_threshold: gr.inputs.Slider = 0.25,
|
48 |
iou_threshold: gr.inputs.Slider = 0.45,
|
49 |
):
|
50 |
+
model_path = download_from_hub(model_path)
|
51 |
model = YOLO(model_path)
|
52 |
|
53 |
model.conf = conf_threshold
|