sourav520 commited on
Commit
35a4532
·
verified ·
1 Parent(s): e51d317

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -4
app.py CHANGED
@@ -133,8 +133,7 @@ def inference(image):
133
  im_rgba = im_rgb.copy()
134
  im_rgba.putalpha(pil_mask)
135
 
136
- return [im_rgba]
137
- #, pil_mask]
138
 
139
 
140
  title = ""
@@ -144,8 +143,7 @@ article = ""
144
  interface = gr.Interface(
145
  fn=inference,
146
  inputs=gr.Image(type='filepath'),
147
- outputs=[gr.Image(type='filepath', format="png")],
148
- examples=[['robot.png'], ['ship.png']],
149
  title=title,
150
  description=description,
151
  article=article,
 
133
  im_rgba = im_rgb.copy()
134
  im_rgba.putalpha(pil_mask)
135
 
136
+ return [im_rgba, pil_mask]
 
137
 
138
 
139
  title = ""
 
143
  interface = gr.Interface(
144
  fn=inference,
145
  inputs=gr.Image(type='filepath'),
146
+ outputs=[gr.Image(type='filepath', format="png"), gr.Image(type='filepath', format="png")],
 
147
  title=title,
148
  description=description,
149
  article=article,