test `output_gallery_component` as output

app.py

@@ -72,7 +72,7 @@ def process(input_image_editor, input_text, progress=gr.Progress(track_tqdm=True
         height=height,
         strength=0.7,
         num_inference_steps=24
-    ).images[0]
+    ).images[0], resized_mask
 
 
 with gr.Blocks() as demo:
@@ -97,7 +97,7 @@ with gr.Blocks() as demo:
                 submit_button_component = gr.Button(
                     value='Submit', variant='primary', scale=0)
         with gr.Column():
-            output_image_component = gr.Image(
+            output_gallery_component = gr.Image(
                 type='pil', image_mode='RGB', label='Generated image')
 
     submit_button_component.click(
@@ -107,7 +107,7 @@ with gr.Blocks() as demo:
             input_text_component
         ],
         outputs=[
-            output_image_component
+            output_gallery_component
         ]
     )