Update app.py

app.py

@@ -310,7 +310,7 @@ with gr.Blocks(css=css) as demo:
                         seed = gr.Number(label="Seed", value=-1)
 
                 with gr.Column(scale=2):
-                    image_output = gr.Gallery(show_label=False, rows=2)
+                    image_output = gr.Gallery(show_label=False, rows=2, preview=True)
                     send_to_img2img = gr.Button(value="Send OUTPUT IMAGE to img2img")
                     send_to_png = gr.Button(value="Send OUTPUT IMAGE to PNG Info")
                     past_url = gr.Textbox(visible=False, interactive=False)
@@ -356,7 +356,7 @@ with gr.Blocks(css=css) as demo:
                         
 
                 with gr.Column(scale=2):
-                    i2i_image_output = gr.Gallery(show_label=False, rows=2)
+                    i2i_image_output = gr.Gallery(show_label=False, rows=2, preview=True)
                     send_to_png_i2i = gr.Button(value="Send INPUT IMAGE to PNG Info")
                     i2i_past_url = gr.Textbox(visible=False, interactive=False)
 
@@ -444,7 +444,7 @@ with gr.Blocks(css=css) as demo:
                         hf_seed = gr.Number(label="Seed", value=-1)
 
                 with gr.Column(scale=2):
-                    hf_image_output = gr.Gallery(show_label=False, preview=True, columns=4, allow_preview=True)
+                    hf_image_output = gr.Gallery(show_label=False, preview=True, rows=2, allow_preview=True)
                     #hf_send_to_img2img = gr.Button(value="Send to img2img")
 
             hf_text_button.click(hf_inference, inputs=[hf_prompt, hf_negative_prompt, hf_model, hf_steps, sampler, hf_cfg_scale, hf_width, hf_height,
@@ -489,7 +489,7 @@ with gr.Blocks(css=css) as demo:
             us_radio = gr.Radio(label="Upscale Levels", choices=[2, 4, 6, 8, 10], value=2, scale=5)
             us_button = gr.Button(value="Generate with Upscaler", variant="primary", scale=5)
             with gr.Column(scale=1):
-                us_output = gr.Gallery(show_label=False, rows=3, preview=True, scale=1)
+                us_output = gr.Gallery(show_label=False, rows=2, preview=True, scale=1)
             send_to_img2img_us = gr.Button(value="Send UPSCALED IMAGE to img2img")
             us_past = gr.Image(interactive=False, visible=False)