Update app.py

app.py

@@ -102,6 +102,9 @@ def run_lora(prompt, cfg_scale, steps, selected_index, randomize_seed, seed, wid
     else:
         prompt_mash = prompt
 
+    with calculateDuration("Unloading LoRA"):
+        pipe.unload_lora_weights()
+        
     # Load LoRA weights
     with calculateDuration(f"Loading LoRA weights for {selected_lora['title']}"):
         if "weights" in selected_lora:
@@ -126,8 +129,6 @@ def run_lora(prompt, cfg_scale, steps, selected_index, randomize_seed, seed, wid
         yield image, seed, gr.update(value=progress_bar, visible=True)
         
     yield final_image, seed, gr.update(value=progress_bar, visible=False)
-    with calculateDuration("Unloading LoRA"):
-        pipe.unload_lora_weights()
 
 def get_huggingface_safetensors(link):
   split_link = link.split("/")