Update app.py

app.py

@@ -165,12 +165,13 @@ with gr.Blocks() as demo:
     """)
     with gr.Row():
         image_input = gr.Image(label="[Stroke] Draw on Image", tool='sketch',type='pil')
+    with gr.Row():
         image_output1 = gr.Image(type="pil", label="Mask with Image")
         image_output2 = gr.Image(type="pil", label="Mask")
         
     image_button = gr.Button("Genarate-Segment-Mask")
 
-    image_button.click(inference_scribble, inputs=image_input, outputs=image_output1)
+    image_button.click(inference_scribble, inputs=image_input, outputs=[image_output1, image_output2])
     image_input.upload(reset_data)
 
 demo.launch()