support mask

app.py

@@ -79,10 +79,11 @@ with demo_blocks as demo:
                             inpaint_mode = gr.Dropdown(label="modes", choices=modes, value="Telea") 
                 with gr.Column():
                     image_out = gr.Image(sources=[],label="Output", elem_id="output-img")
+                    mask_out = gr.Image(height=800,sources=[],label="Mask", elem_id="mask-img",format="jpeg")
                     
             
 
-    btn.click(fn=process_images, inputs=[image, image_mask,inpaint_radius,blur_radius,edge_expand,inpaint_mode], outputs =image_out, api_name='infer')
+    btn.click(fn=process_images, inputs=[image, image_mask,inpaint_radius,blur_radius,edge_expand,inpaint_mode], outputs =[image_out,mask_out], api_name='infer')
     gr.Examples(
                examples=[["examples/street.jpg", "examples/street_mask.jpg"]]
 ,