Update app.py

app.py

@@ -405,14 +405,14 @@ def run_lora(prompt, cfg_scale, steps, selected_indices, lora_scale_1, lora_scal
 #    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
+#        final_image = None
+#        step_counter = 0
         for image in image_generator:
             step_counter += 1
-            final_image = image
+#            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)
+#        yield final_image, seed, gr.update(value=progress_bar, visible=False)
 
 run_lora.zerogpu = True