Muhammad Taqi Raza
commited on
Commit
·
376cb79
1
Parent(s):
1b37cad
add app files
Browse files
app.py
CHANGED
|
@@ -9,14 +9,14 @@ from huggingface_hub import snapshot_download
|
|
| 9 |
# Step 1: Download Model Weights
|
| 10 |
# ----------------------------------------
|
| 11 |
MODEL_REPO = "roll-ai/DOVE"
|
| 12 |
-
MODEL_PATH = "pretrained_models/"
|
| 13 |
|
| 14 |
if not os.path.exists(MODEL_PATH) or len(os.listdir(MODEL_PATH)) == 0:
|
| 15 |
print("🔽 Downloading model weights from Hugging Face Hub...")
|
| 16 |
snapshot_download(
|
| 17 |
repo_id=MODEL_REPO,
|
| 18 |
repo_type="dataset",
|
| 19 |
-
local_dir=
|
| 20 |
local_dir_use_symlinks=False
|
| 21 |
)
|
| 22 |
print("✅ Download complete.")
|
|
|
|
| 9 |
# Step 1: Download Model Weights
|
| 10 |
# ----------------------------------------
|
| 11 |
MODEL_REPO = "roll-ai/DOVE"
|
| 12 |
+
MODEL_PATH = "pretrained_models/DOVE"
|
| 13 |
|
| 14 |
if not os.path.exists(MODEL_PATH) or len(os.listdir(MODEL_PATH)) == 0:
|
| 15 |
print("🔽 Downloading model weights from Hugging Face Hub...")
|
| 16 |
snapshot_download(
|
| 17 |
repo_id=MODEL_REPO,
|
| 18 |
repo_type="dataset",
|
| 19 |
+
local_dir="pretrained_models/",
|
| 20 |
local_dir_use_symlinks=False
|
| 21 |
)
|
| 22 |
print("✅ Download complete.")
|