Update app.py

app.py

@@ -61,7 +61,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 @@ for ratio in PRUNING_RATIOS:
 print("📥 Preloading LoRA checkpoint for 20% pruning ratio...")
 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_20/*"]
     )
     lora_weights = load_file(os.path.join(lora_repo_path, "dev", "lora", "prune_20", LORA_WEIGHT_NAME_SAFE))