Update app.py

app.py

@@ -59,7 +59,7 @@ for ratio in PRUNING_RATIOS:
     try:
         print(f"Loading {ratio}% pruned model...")
         model_file = hf_hub_download(
-            repo_id="LWZ19/sdxl_prune",
+            repo_id="LWZ19/ecodiff_sdxl_prune",
             filename=f"pruned_model_{ratio}.pkl"
         )
         
@@ -83,7 +83,7 @@ for ratio in PRUNING_RATIOS:
     try:
         print(f"Loading {ratio}% LoRA weights...")
         lora_file = hf_hub_download(
-            repo_id="LWZ19/sdxl_retrain_weights",
+            repo_id="LWZ19/ecodiff_sdxl_retrain_weights",
             filename=f"lora/prune_{ratio}/{LORA_WEIGHT_NAME_SAFE}"
         )
         lora_weights[ratio] = os.path.dirname(lora_file)
@@ -96,7 +96,7 @@ for ratio in PRUNING_RATIOS:
     try:
         print(f"Loading {ratio}% full weights...")
         full_file = hf_hub_download(
-            repo_id="LWZ19/sdxl_retrain_weights",
+            repo_id="LWZ19/ecodiff_sdxl_retrain_weights",
             filename=f"full/prune_{ratio}/unet/{SAFETENSORS_WEIGHTS_NAME}"
         )
         full_weights[ratio] = load_file(full_file)