Spaces:
Running
on
CPU Upgrade
Running
on
CPU Upgrade
Update app.py
Browse files
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 |
-
|
71 |
repo_id=repo_name,
|
72 |
local_dir="/data",
|
73 |
-
|
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"
|