Spaces:
Running
Running
Commit
·
2fed831
1
Parent(s):
70125a1
Migrate model
Browse files- download_model.py +1 -1
download_model.py
CHANGED
@@ -10,7 +10,7 @@ print("⏳ Downloading the SentenceTransformer model...")
|
|
10 |
model_path = snapshot_download(repo_id=MODEL_REPO, cache_dir=MODEL_CACHE_DIR)
|
11 |
|
12 |
# Find the correct snapshot directory
|
13 |
-
snapshots_dir = os.path.join(model_path, "snapshots
|
14 |
if os.path.exists(snapshots_dir):
|
15 |
snapshot_folders = os.listdir(snapshots_dir)
|
16 |
if snapshot_folders:
|
|
|
10 |
model_path = snapshot_download(repo_id=MODEL_REPO, cache_dir=MODEL_CACHE_DIR)
|
11 |
|
12 |
# Find the correct snapshot directory
|
13 |
+
snapshots_dir = os.path.join(model_path, "snapshots") # snapshot
|
14 |
if os.path.exists(snapshots_dir):
|
15 |
snapshot_folders = os.listdir(snapshots_dir)
|
16 |
if snapshot_folders:
|