Update app.py

app.py

@@ -395,6 +395,7 @@ def generate(
     steps=9,
     shift=3.0,
     random_seed=True,
+    return_image=True,
     gallery_images=None,
     enhance=False,
     progress=gr.Progress(track_tqdm=True),
@@ -455,8 +456,8 @@ def generate(
 
     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
+    if return_image:
+        gallery_images = [image] + gallery_images # latest output to be at the top of the list
 
     timestamp = int(time.time())
     filename = f"z_generated_{timestamp}_{seed}.png"
@@ -507,6 +508,7 @@ with gr.Blocks(title="Z-Image Demo") as demo:
             with gr.Row():
                 seed = gr.Number(label="Seed", value=42, precision=0)
                 random_seed = gr.Checkbox(label="Random Seed", value=True)
+            return_image = gr.Checkbox(label="返回图片", value=True)
 
             with gr.Row():
                 steps = gr.Slider(label="Steps", minimum=1, maximum=100, value=8, step=1, interactive=False)
@@ -553,7 +555,7 @@ with gr.Blocks(title="Z-Image Demo") as demo:
 
     generate_btn.click(
         generate,
-        inputs=[prompt_input, resolution, seed, steps, shift, random_seed, output_gallery],
+        inputs=[prompt_input, resolution, seed, steps, shift, random_seed, return_image, output_gallery],
         outputs=[output_gallery, used_seed, seed, image_path_output],
         api_visibility="public",
     )