Update app.py
Browse files
app.py
CHANGED
@@ -67,7 +67,14 @@ for f in rm_files:
|
|
67 |
shutil.rmtree(f)
|
68 |
|
69 |
if not os.path.exists(final_model_path):
|
70 |
-
snapshot_download(
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
71 |
# snapshot_download(repo_id=embedder_name, local_dir="/data/paraphrase-multilingual-mpnet-base-v2", cache_dir="/data")
|
72 |
print("Files downloaded!")
|
73 |
|
|
|
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"
|
77 |
+
)
|
78 |
# snapshot_download(repo_id=embedder_name, local_dir="/data/paraphrase-multilingual-mpnet-base-v2", cache_dir="/data")
|
79 |
print("Files downloaded!")
|
80 |
|