Update app.py

app.py

@@ -158,6 +158,18 @@ def generate(
 
 
 # ==================== Gradio Interface ====================
+output_gallery = gr.Gallery(
+    label="Generated Images",
+    columns=2,
+    rows=2,
+    height=600,
+    object_fit="contain",
+    format="png",
+    interactive=False,
+)
+
+used_seed = gr.Textbox(label="Seed Used", interactive=False)
+
 with gr.Blocks(title="Z-Image Demo") as demo:
     gr.Markdown(
         """<div align="center">
@@ -241,16 +253,8 @@ with gr.Blocks(title="Z-Image Demo") as demo:
             )
 
         with gr.Column(scale=1):
-            output_gallery = gr.Gallery(
-                label="Generated Images",
-                columns=2,
-                rows=2,
-                height=600,
-                object_fit="contain",
-                format="png",
-                interactive=False,
-            )
-            used_seed = gr.Textbox(label="Seed Used", interactive=False)
+            output_gallery.render()
+            used_seed.render()
 
     # Event handlers
     def update_res_choices(res_cat_value):