Update app.py

app.py

@@ -139,16 +139,14 @@ def run_lora(prompt, image_input, image_strength, cfg_scale, steps, selected_ind
         
     # Load LoRA weights
     with calculateDuration(f"Loading LoRA weights for {selected_lora['title']}"):
-        if(image_input is not None):
-            if "weights" in selected_lora:
-                pipe_i2i.load_lora_weights(lora_path, weight_name=selected_lora["weights"])
-            else:
-                pipe_i2i.load_lora_weights(lora_path)
-        else:
-            if "weights" in selected_lora:
-                pipe.load_lora_weights(lora_path, weight_name=selected_lora["weights"])
-            else:
-                pipe.load_lora_weights(lora_path)
+        pipe_to_use = pipe_i2i if image_input is not None else pipe
+        weight_name = selected_lora.get("weights", None)
+        
+        pipe_to_use.load_lora_weights(
+            lora_path, 
+            weight_name=weight_name, 
+            low_cpu_mem_usage=False
+        )
                 
     # Set random seed for reproducibility
     with calculateDuration("Randomizing seed"):