separate image output component for mask

app.py

@@ -143,14 +143,10 @@ with gr.Blocks() as demo:
                         value=20,
                     )
         with gr.Column():
-            output_gallery_component = gr.Gallery(
-                columns=[1],
-                rows=[2],
-                selected_index=0,
-                preview=True,
-                object_fit="contain",
-                height="auto",
-                label='Generated image')
+            output_image_component = gr.Image(
+                type='pil', image_mode='RGB', label='Generated image')
+            output_mask_component = gr.Image(
+                type='pil', image_mode='RGB', label='Input mask')
 
     submit_button_component.click(
         fn=process,
@@ -163,7 +159,8 @@ with gr.Blocks() as demo:
             num_inference_steps_slider_component
         ],
         outputs=[
-            output_gallery_component
+            output_image_component,
+            output_mask_component
         ]
     )