IlyaGusev commited on
Commit
9d963de
1 Parent(s): 1b64878

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -67,10 +67,10 @@ for f in rm_files:
67
  shutil.rmtree(f)
68
 
69
  if not os.path.exists(final_model_path):
70
- snapshot_download(
71
  repo_id=repo_name,
72
  local_dir="/data",
73
- allow_patterns=model_name,
74
  local_dir_use_symlinks=False,
75
  force_download=True,
76
  cache_dir="/data"
 
67
  shutil.rmtree(f)
68
 
69
  if not os.path.exists(final_model_path):
70
+ hf_hub_download(
71
  repo_id=repo_name,
72
  local_dir="/data",
73
+ filename=model_name,
74
  local_dir_use_symlinks=False,
75
  force_download=True,
76
  cache_dir="/data"