rahul7star commited on
Commit
e4b3867
·
verified ·
1 Parent(s): 9a419b5

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -324,7 +324,7 @@ def fetch_images_and_generate_captions():
324
  # Download all files from the dataset repo
325
  snapshot_path = snapshot_download(
326
  repo_id=REPO_ID,
327
- repo_type="dataset",
328
  local_dir=local_dir,
329
  local_dir_use_symlinks=False,
330
  allow_patterns=[f"{FOLDER_IN_REPO}/*"], # only files inside the subfolder
@@ -509,7 +509,7 @@ def auto_run_lora_from_repo():
509
 
510
  snapshot_path = snapshot_download(
511
  repo_id=REPO_ID,
512
- repo_type="dataset",
513
  local_dir=local_dir,
514
  local_dir_use_symlinks=False,
515
  allow_patterns=[f"{FOLDER_IN_REPO}/*"], # only files inside the subfolder
 
324
  # Download all files from the dataset repo
325
  snapshot_path = snapshot_download(
326
  repo_id=REPO_ID,
327
+ repo_type="model",
328
  local_dir=local_dir,
329
  local_dir_use_symlinks=False,
330
  allow_patterns=[f"{FOLDER_IN_REPO}/*"], # only files inside the subfolder
 
509
 
510
  snapshot_path = snapshot_download(
511
  repo_id=REPO_ID,
512
+ repo_type="model",
513
  local_dir=local_dir,
514
  local_dir_use_symlinks=False,
515
  allow_patterns=[f"{FOLDER_IN_REPO}/*"], # only files inside the subfolder