Update app.py

app.py

@@ -134,7 +134,7 @@ def inference(image: Image):
   im_rgba = im_rgb.copy()
   im_rgba.putalpha(pil_mask)
 
-  return [im_rgba]
+  return [im_rgba,pil_mask]
 
 
 title = ""
@@ -144,7 +144,7 @@ article = ""
 interface = gr.Interface(
     fn=inference,
     inputs=gr.Image(type='filepath'),
-    outputs=["image"],
+    outputs=["image","image"],
     examples=[['robot.png'], ['ship.png']],
     title=title,
     description=description,