Adjust the layout

app.py

@@ -337,7 +337,7 @@ if __name__ == "__main__":
                         type='pil',
                         elem_id="img2img_mask",
                         image_mode="RGBA"
-                    )
+                    ).style(height=300)
                     img2img_image_editor = gr.Image(
                         value=sample_img2img,
                         source="upload",
@@ -379,7 +379,7 @@ if __name__ == "__main__":
                 gr.Markdown('#### 编辑后的图片')
                 with gr.Row():
                     output_img2img_gallery = gr.Gallery(label="Images", elem_id="img2img_gallery_output").style(
-                        grid=[1,1])
+                        grid=[4,4,4] )
                     img2img_job_ui = job_manager.draw_gradio_ui() if job_manager else None
                     with gr.Column(visible=False):
                         with gr.Tabs(visible=False):