Journey9ni commited on
Commit
116d5b4
·
1 Parent(s): 50a3b36

Update snapshot_download call to specify local directory for model files.

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -31,7 +31,7 @@ model_path_map = {
31
  }
32
  os.makedirs(local_dir, exist_ok=True)
33
  # download remote repo
34
- snapshot_download(repo_id=repo_id)
35
 
36
  # rename the files
37
  for remote_name, local_name in model_path_map.items():
 
31
  }
32
  os.makedirs(local_dir, exist_ok=True)
33
  # download remote repo
34
+ snapshot_download(repo_id=repo_id, local_dir='./')
35
 
36
  # rename the files
37
  for remote_name, local_name in model_path_map.items():