Leon Sick commited on
Commit
5cd43b7
·
1 Parent(s): af3c6d0

start of space

Browse files
Files changed (1) hide show
  1. model.py +2 -1
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
- subprocess.run(shlex.split(f'huggingface-cli download leonsick/cuts3d_zeroshot {model_filename} --token {hf_token} --local-dir {model_dir}'))
 
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}'))