Spaces:
Runtime error
Runtime error
valentynliubchenko commited on
Commit ·
f122098
1
Parent(s): 02e3ba7
fixed path
Browse files
app.py
CHANGED
|
@@ -6,7 +6,7 @@ import numpy as np
|
|
| 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 |
|
| 11 |
def process_image(input_image, model_name, conf):
|
| 12 |
print(model_name)
|
|
|
|
| 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 |
|
| 11 |
def process_image(input_image, model_name, conf):
|
| 12 |
print(model_name)
|