Update app.py

app.py

@@ -64,11 +64,8 @@ with block:
                 )
                 btn = gr.Button("Generate image")
         gallery = gr.Gallery(
-            label="Generated images", show_label=False, elem_id="gallery", width = 768
-        )
-        # gallery = gr.Image(
-        #     label="Generated images", elem_id="gallery", width = 768, height = 536
-        # )
+            label="Generated images", show_label=False, elem_id="gallery", width = 512
+        ).style(columns=[2], rows=[2], object_fit="contain", height="auto")
 
 
         with gr.Row(elem_id="advanced-options"):