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

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -2
app.py CHANGED
@@ -133,7 +133,8 @@ def inference(image):
133
  im_rgba = im_rgb.copy()
134
  im_rgba.putalpha(pil_mask)
135
 
136
- return [im_rgba, pil_mask]
 
137
 
138
 
139
  title = ""
@@ -143,7 +144,7 @@ article = ""
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
  examples=[['robot.png'], ['ship.png']],
148
  title=title,
149
  description=description,
 
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
  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,