Update app.py

app.py

@@ -430,14 +430,14 @@ def remove_custom_lora(selected_indices, current_loras, gallery):
         lora_image_4 = lora4['image']
     return (current_loras, gr.update(value=gallery_items), selected_info_1, selected_info_2, selected_info_3, selected_info_4, selected_indices, lora_scale_1, lora_scale_2, lora_scale_3, lora_scale_4, lora_image_1, lora_image_2, lora_image_3, lora_image_4)
 
-def generate_image(prompt_mash, steps, seed, cfg_scale, width, height, progress):
+def generate_image(prompt, steps, seed, cfg_scale, width, height, progress):
     print("Generating image...")
     pipe.to("cuda")
     generator = torch.Generator(device="cuda").manual_seed(seed)
     with calculateDuration("Generating image"):
         # Generate image
         for img in pipe.flux_pipe_call_that_returns_an_iterable_of_images(
-            prompt=prompt_mash,
+            prompt=prompt,
             num_inference_steps=steps,
             guidance_scale=cfg_scale,
             width=width,
@@ -447,7 +447,8 @@ def generate_image(prompt_mash, steps, seed, cfg_scale, width, height, progress)
             output_type="pil",
             good_vae=good_vae,
         ):
-            yield img
+            # Yielding a tuple with image, seed, and a progress update
+            yield img, seed, f"Generated image {img} with seed {seed}"
     return img
 
 @spaces.GPU(duration=75)