Update app.py

app.py

@@ -185,7 +185,7 @@ with block:
             vae_encoder_tile_size = gr.Slider(label="Encoder tile size", minimum=512, maximum=5000, value=1024, step=256, info="Size of each tile for the VAE encoder.")
             vae_decoder_tile_size = gr.Slider(label="Decoder tile size", minimum=64, maximum=512, value=224, step=128, info="Size of each tile for the VAE decoder.")
     with gr.Column():
-        result_gallery = gr.Gallery(label="Output", show_label=False, elem_id="gallery").style(grid=2, height="auto")
+        result_gallery = gr.Gallery(label="Output", show_label=False, elem_id="gallery")
 
     inputs = [
         input_image,