Muhammad Taqi Raza commited on
Commit
376cb79
·
1 Parent(s): 1b37cad

add app files

Browse files
Files changed (1) hide show
  1. app.py +2 -2
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=MODEL_PATH,
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.")