Update app.py

app.py

@@ -384,8 +384,8 @@ def run_lora(prompt, cfg_scale, steps, selected_indices, lora_scale_1, lora_scal
                 low_cpu_mem_usage=True,
                 adapter_name=lora_name
             )
-#            if image_input is not None: pipe_i2i = pipe_to_use
-#            else: pipe = pipe_to_use
+            if image_input is not None: pipe_i2i = pipe_to_use
+            else: pipe = pipe_to_use
         print("Loaded LoRAs:", lora_names)
         print("Adapter weights:", lora_weights)
         if image_input is not None:
@@ -398,21 +398,21 @@ def run_lora(prompt, cfg_scale, steps, selected_indices, lora_scale_1, lora_scal
         if randomize_seed:
             seed = random.randint(0, MAX_SEED)
 
-    # Generate image
-    #if image_input is not None:
+# Generate image
+    if image_input is not None:
         final_image = generate_image_to_image(prompt_mash, steps, cfg_scale, width, height, seed)
         yield final_image, seed, gr.update(visible=False)
-        else:
-            image_generator = generate_image(prompt_mash, steps, seed, cfg_scale, width, height, progress)
-        # Consume the generator to get the final image
-            final_image = None
-            step_counter = 0
-            for image in image_generator:
-                step_counter += 1
-                final_image = image
-                progress_bar = f'<div class="progress-container"><div class="progress-bar" style="--current: {step_counter}; --total: {steps};"></div></div>'
-                yield image, seed, gr.update(value=progress_bar, visible=True)
-            yield final_image, seed, gr.update(value=progress_bar, visible=False)
+    else:
+        image_generator = generate_image(prompt_mash, steps, seed, cfg_scale, width, height, progress)
+# Consume the generator to get the final image
+        final_image = None
+        step_counter = 0
+        for image in image_generator:
+            step_counter += 1
+            final_image = image
+            progress_bar = f'<div class="progress-container"><div class="progress-bar" style="--current: {step_counter}; --total: {steps};"></div></div>'
+            yield image, seed, gr.update(value=progress_bar, visible=True)
+        yield final_image, seed, gr.update(value=progress_bar, visible=False)
 
 run_lora.zerogpu = True