Leon Sick
commited on
Commit
•
f7a79fa
1
Parent(s):
1726cd1
start of space
Browse files
model.py
CHANGED
@@ -90,6 +90,8 @@ def load_model(score_threshold: float) -> VisualizationDemo:
|
|
90 |
model_dir = pathlib.Path('checkpoints')
|
91 |
model_dir.mkdir(exist_ok=True)
|
92 |
weight_path = model_dir / WEIGHT_URL.split('/')[-1]
|
|
|
|
|
93 |
if not weight_path.exists():
|
94 |
subprocess.run(shlex.split(f'wget {WEIGHT_URL} -O {weight_path}'))
|
95 |
|
|
|
90 |
model_dir = pathlib.Path('checkpoints')
|
91 |
model_dir.mkdir(exist_ok=True)
|
92 |
weight_path = model_dir / WEIGHT_URL.split('/')[-1]
|
93 |
+
print("***weight_path***", weight_path)
|
94 |
+
|
95 |
if not weight_path.exists():
|
96 |
subprocess.run(shlex.split(f'wget {WEIGHT_URL} -O {weight_path}'))
|
97 |
|