Update app.py

app.py

@@ -328,7 +328,7 @@ def generate_image(prompt_mash, steps, seed, cfg_scale, width, height, progress)
 #        joint_attention_kwargs={"scale": 1.0},
 #        output_type="pil",
 #    ).images[0]
-    return final_image
+    return image
 
 @spaces.GPU(duration=75)
 def run_lora(prompt, cfg_scale, steps, selected_indices, lora_scale_1, lora_scale_2, randomize_seed, seed, width, height, loras_state, progress=gr.Progress(track_tqdm=True)):
@@ -402,7 +402,7 @@ def run_lora(prompt, cfg_scale, steps, selected_indices, lora_scale_1, lora_scal
             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