Update app.py

app.py

@@ -35,7 +35,7 @@ import json
 import yaml
 from slugify import slugify
 
-from flux_train_ui import start_training
+from flux_train_ui import start_training 
 
 
 
@@ -552,14 +552,12 @@ def auto_run_lora_from_repo():
         local_dir = Path(f"/tmp/{LORA_NAME}-{uuid.uuid4()}")
         os.makedirs(local_dir, exist_ok=True)
 
-        hf_hub_download(
-            repo_id=REPO_ID,
-            repo_type="dataset",
-            filename=".placeholder",  # needs a dummy filename to trigger folder download
-            subfolder=FOLDER_IN_REPO,
-            local_dir=local_dir,
-            local_dir_use_symlinks=False,
-            force_download=False
+        snapshot_path = snapshot_download(
+          repo_id=REPO_ID,
+          repo_type="model",
+          local_dir=local_dir,
+          local_dir_use_symlinks=False,
+           allow_patterns=[f"{FOLDER_IN_REPO}/*"],  # only files inside the subfolder
         )
 
         image_dir = local_dir / FOLDER_IN_REPO