Update app.py

app.py

@@ -41,7 +41,7 @@ def infer(prompt, negative_prompt, guidance_scale=10, num_inference_steps=50):
     return imgs
     
 # output = gr.Gallery(label="Outputs").style(grid=(1,2))
-output = gr.Gallery(label="Outputs")#, columns=2, rows=1)
+output = gr.Gallery(label="Outputs", rows=1, columns=4, object_fit="contain", height="auto")#, columns=2, rows=1)
 
 # customize interface
 title = "KerasCV Stable Diffusion Demo on images of Ace."
@@ -176,7 +176,7 @@ with gr.Blocks() as demo:
             prompt_steps = gr.Slider(label="Inference Steps",value=50)
             prompt_btn = gr.Button("Generate")
         with gr.Column():
-            output = gr.Gallery(label="Outputs")#.style(grid=(1,2))
+            output = gr.Gallery(label="Outputs", object_fit="contain", height="auto")#.style(grid=(1,2))
     prompt_btn.click(infer, inputs=[prompt_pos, prompt_neg, prompt_gs, prompt_steps], outputs=[output])
     with gr.Row():
         gr.Examples(examples, inputs=[prompt_pos, prompt_neg, prompt_gs, prompt_steps], outputs=output, fn=infer, cache_examples=True)