Update app.py

app.py

@@ -60,7 +60,7 @@ for ratio in PRUNING_RATIOS:
     try:
         print(f"Loading {ratio}% pruned model...")
         model_file = hf_hub_download(
-            repo_id="LWZ19/flux_prune",
+            repo_id="LWZ19/ecodiff_flux_prune",
             filename=f"dev/pruned_model_{ratio}.pkl"
         )
         
@@ -79,7 +79,7 @@ print("📥 Preloading all LoRA weights...")
 for ratio in PRUNING_RATIOS:
     try:
         lora_repo_path = snapshot_download(
-            repo_id="LWZ19/flux_retrain_weights",
+            repo_id="LWZ19/ecodiff_flux_retrain_weights",
             allow_patterns=[f"dev/lora/prune_{ratio}/*"]
         )
         lora_weights = load_file(os.path.join(lora_repo_path, "dev", "lora", f"prune_{ratio}", LORA_WEIGHT_NAME_SAFE))