LogicGoInfotechSpaces commited on
Commit
5050e58
·
1 Parent(s): 4a63fe7

Fix file path issue - download to current directory instead of nested pretrained_models

Browse files
Files changed (1) hide show
  1. app.py +1 -2
app.py CHANGED
@@ -47,8 +47,7 @@ def ensure_weights():
47
  try:
48
  snapshot_download(
49
  repo_id=repo_id,
50
- local_dir="pretrained_models",
51
- local_dir_use_symlinks=False,
52
  allow_patterns=["**/*"],
53
  token=os.getenv("HF_TOKEN") or os.getenv("HUGGINGFACE_TOKEN"),
54
  )
 
47
  try:
48
  snapshot_download(
49
  repo_id=repo_id,
50
+ local_dir=".",
 
51
  allow_patterns=["**/*"],
52
  token=os.getenv("HF_TOKEN") or os.getenv("HUGGINGFACE_TOKEN"),
53
  )