Update app.py

app.py

@@ -238,9 +238,10 @@ def run_lora(prompt, image_input, image_strength, cfg_scale, steps, selected_ind
                     pipe.load_lora_weights(lora_path, weight_name=lora["weights"], low_cpu_mem_usage=True, adapter_name=lora_name)
                 else:
                     pipe.load_lora_weights(lora_path, low_cpu_mem_usage=True, adapter_name=lora_name)
-
-    pipeline.set_adapters(lora_names, adapter_weights=[lora_scale_1, lora_scale_2])
-    
+        if image_input is not None:
+            pipe_i2i.set_adapters(lora_names, adapter_weights=[lora_scale_1, lora_scale_2])
+        else:
+            pipe.set_adapters(lora_names, adapter_weights=[lora_scale_1, lora_scale_2])
     # Set random seed for reproducibility
     with calculateDuration("Randomizing seed"):
         if randomize_seed: