wahaha commited on
Commit
86527b0
·
1 Parent(s): 9f2bb03
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -139,7 +139,7 @@ def process(im):
139
 
140
  print(results)
141
 
142
- return Image.open(results[0]), Image.open(im.name)
143
 
144
  title = "U-2-Net"
145
  description = "Gradio demo for U-2-Net, https://github.com/xuebinqin/U-2-Net"
@@ -149,7 +149,7 @@ gr.Interface(
149
  process,
150
  [gr.inputs.Image(type="file", label="Input")
151
  ],
152
- [gr.outputs.Image(type="pil", label="Output"), gr.outputs.Image(type="pil", label="Output")],
153
  title=title,
154
  description=description,
155
  article=article,
 
139
 
140
  print(results)
141
 
142
+ return Image.open(results[0])
143
 
144
  title = "U-2-Net"
145
  description = "Gradio demo for U-2-Net, https://github.com/xuebinqin/U-2-Net"
 
149
  process,
150
  [gr.inputs.Image(type="file", label="Input")
151
  ],
152
+ [gr.outputs.Image(type="pil", label="Output")],
153
  title=title,
154
  description=description,
155
  article=article,