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}'))
|