Update app.py

app.py

@@ -562,7 +562,7 @@ with gr.Blocks(title="Z-Image Demo") as demo:
     generate_btn.click(
         generate,
         inputs=[prompt_input, resolution, seed, steps, shift, random_seed],
-        outputs=[image, used_seed, seed],
+        outputs=[output_gallery, used_seed, seed],
         api_visibility="public",
     )