Spaces:
Running
Running
Update app.py
Browse files
app.py
CHANGED
@@ -10,7 +10,7 @@ model_path = "C:/Users/KimBok/Downloads/best_int8_openvino_model"
|
|
10 |
def load_model(repo_id):
|
11 |
download_dir = snapshot_download(repo_id)
|
12 |
print(download_dir)
|
13 |
-
path = os.path.join(download_dir, "
|
14 |
print(path)
|
15 |
detection_model = YOLO(path, task='detect')
|
16 |
return detection_model
|
|
|
10 |
def load_model(repo_id):
|
11 |
download_dir = snapshot_download(repo_id)
|
12 |
print(download_dir)
|
13 |
+
path = os.path.join(download_dir, "best_int8_openvino_model")
|
14 |
print(path)
|
15 |
detection_model = YOLO(path, task='detect')
|
16 |
return detection_model
|