Update app.py

app.py

@@ -398,7 +398,6 @@ def generate(
     steps=9,
     shift=3.0,
     random_seed=True,
-    gallery_images=None,
     enhance=False,
     progress=gr.Progress(track_tqdm=True),
 ):
@@ -416,13 +415,11 @@ def generate(
         steps (int): Number of inference steps for the diffusion process
         shift (float): Time shift parameter for the flow matching scheduler
         random_seed (bool): Whether to generate a new random seed, if True will ignore the seed input
-        gallery_images (list): List of previously generated images to append to (only needed for the Gradio UI)
         enhance (bool): This was Whether to enhance the prompt (DISABLED! Do not use)
         progress (gr.Progress): Gradio progress tracker for displaying generation progress (only needed for the Gradio UI)
 
     Returns:
         tuple: (gallery_images, seed_str, seed_int)
-            - gallery_images: Updated list of generated images including the new image
             - seed_str: String representation of the seed used for generation
             - seed_int: Integer representation of the seed used for generation
     """
@@ -479,12 +476,8 @@ def generate(
     except UnsafeContentError:
         image = Image.open("nsfw.png")
 
-    if gallery_images is None:
-        gallery_images = []
-    # gallery_images.append(image)
-    gallery_images = [image] + gallery_images # latest output to be at the top of the list
 
-    return gallery_images, str(new_seed), int(new_seed)
+    return image, str(new_seed), int(new_seed)
 
 
 init_app()
@@ -568,8 +561,8 @@ with gr.Blocks(title="Z-Image Demo") as demo:
 
     generate_btn.click(
         generate,
-        inputs=[prompt_input, resolution, seed, steps, shift, random_seed, output_gallery],
-        outputs=[output_gallery, used_seed, seed],
+        inputs=[prompt_input, resolution, seed, steps, shift, random_seed],
+        outputs=[image, used_seed, seed],
         api_visibility="public",
     )