Update app.py

app.py

@@ -150,9 +150,8 @@ with gr.Blocks(theme='gradio/soft') as demo:
             # submit_btn.click(fn=infer, inputs=inputs, outputs=[output_image])
     
         with gr.Column():
-            output_image = gr.Gallery(label='Output Image', show_label=False, elem_id="gallery").style(grid=3, height='auto')
+            output_image = gr.Gallery(label='Output Image', show_label=False, elem_id="gallery").style(grid=2, height='auto')
 
-    gr.Markdown("### Image Examples")
     gr.Examples(
         examples=[
             [