Update app.py

app.py

@@ -82,7 +82,10 @@ with gr.Blocks() as demo:
                     seed = gr.Slider(minimum=0, maximum=1000, step=1, value=42, label='Seed')
                     button = gr.Button("Generate", variant="primary")
                 with gr.Column():
-                    output = gr.Gallery(label='Generated Images', object_fit="scale-down")
+                    output = gr.Gallery(label='Generated Images',
+                                        columns=4,
+                                        rows=1,
+                                        height=256, object_fit="scale-down")
                     button.click(demo_infer, inputs=[
                         guidance_scale, randomize_temperature, num_sample_steps,
                         i1k_class, seed],