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 |
|