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.")
|