Update app.py

app.py

@@ -402,17 +402,17 @@ def run_lora(prompt, cfg_scale, steps, selected_indices, lora_scale_1, lora_scal
     #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)
+        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)
+            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