Leon Sick commited on
Commit ·
5cd43b7
1
Parent(s): af3c6d0
start of space
Browse files
model.py
CHANGED
|
@@ -102,7 +102,8 @@ def load_model(score_threshold: float) -> VisualizationDemo:
|
|
| 102 |
|
| 103 |
|
| 104 |
# Load the model weights file from huggingface hub, use token to authenticate
|
| 105 |
-
|
|
|
|
| 106 |
|
| 107 |
#if not weight_path.exists():
|
| 108 |
# subprocess.run(shlex.split(f'wget {WEIGHT_URL} -O {weight_path}'))
|
|
|
|
| 102 |
|
| 103 |
|
| 104 |
# Load the model weights file from huggingface hub, use token to authenticate
|
| 105 |
+
if not weight_path.exists():
|
| 106 |
+
subprocess.run(shlex.split(f'huggingface-cli download leonsick/cuts3d_zeroshot {model_filename} --token {hf_token} --local-dir {model_dir}'))
|
| 107 |
|
| 108 |
#if not weight_path.exists():
|
| 109 |
# subprocess.run(shlex.split(f'wget {WEIGHT_URL} -O {weight_path}'))
|