Fixed grid on mobile devices

app.py

@@ -110,7 +110,7 @@ cond_img = gr.Image(label="Input", shape=(512, 512), value=default_example[2])\
                     .style(height=200)
 
 output = gr.Gallery(label="Generated images")\
-                    .style(height=200, rows=[2], columns=[1, 2], object_fit="contain")
+                    .style(height=200, rows=[2], columns=[2], object_fit="contain")
 
 prompt = gr.Textbox(lines=1, label="Prompt", value=default_example[0])
 negative_prompt = gr.Textbox(lines=1, label="Negative Prompt", value=default_example[1])