Spaces:
Runtime error
Runtime error
valentynliubchenko
commited on
Commit
·
e2717de
1
Parent(s):
0c6b368
Confidence Threshold
Browse files
app.py
CHANGED
|
@@ -4,7 +4,7 @@ from ultralytics import YOLO
|
|
| 4 |
import numpy as np
|
| 5 |
|
| 6 |
model_options = ["xViewyolov8n_v8_100e.pt", "xViewyolov8s_v8_100e.pt", "xViewyolov8m_v8_100e.pt"]
|
| 7 |
-
model_names = ["Nano", "
|
| 8 |
models = [YOLO(os.path.join("./saved_model", option)) for option in model_options]
|
| 9 |
example_list = [["examples/" + example] for example in os.listdir("examples")]
|
| 10 |
|
|
|
|
| 4 |
import numpy as np
|
| 5 |
|
| 6 |
model_options = ["xViewyolov8n_v8_100e.pt", "xViewyolov8s_v8_100e.pt", "xViewyolov8m_v8_100e.pt"]
|
| 7 |
+
model_names = ["Nano", "Small", "Medium"]
|
| 8 |
models = [YOLO(os.path.join("./saved_model", option)) for option in model_options]
|
| 9 |
example_list = [["examples/" + example] for example in os.listdir("examples")]
|
| 10 |
|