|
@@ -109,6 +109,7 @@ def load_model(model_name):
|
|
|
|
|
|
|
|
# check root of models folder, and model path root
|
|
# check root of models folder, and model path root
|
|
|
paths = [ f"{path_to_model}/{pt_model}", f"models/{pt_model}" ]
|
|
paths = [ f"{path_to_model}/{pt_model}", f"models/{pt_model}" ]
|
|
|
|
|
+ pt_path = None
|
|
|
for path in [ Path(p) for p in paths ]:
|
|
for path in [ Path(p) for p in paths ]:
|
|
|
if path.exists():
|
|
if path.exists():
|
|
|
pt_path = path
|
|
pt_path = path
|