init

app.py

@@ -91,9 +91,10 @@ net.eval()
 
 def process(im):
     image = modnet.segment(im.name)
-    Image.fromarray(np.uint8(image)).save(im.name)
+    im_path = os.path.abspath(os.path.basename(im.name))
+    Image.fromarray(np.uint8(image)).save(im_path)
 
-    img_name_list = [im.name]
+    img_name_list = [im_path]
     print("Number of images: ", len(img_name_list))
     # --------- 2. dataloader ---------
     # 1. dataloader
@@ -134,7 +135,7 @@ def process(im):
 
     print(results)
 
-    return Image.open(results[0]), Image.open(im.name)
+    return Image.open(results[0])
         
 title = "U-2-Net"
 description = "Gradio demo for U-2-Net, https://github.com/xuebinqin/U-2-Net"
@@ -144,7 +145,7 @@ gr.Interface(
     process, 
     [gr.inputs.Image(type="file", label="Input")
 ], 
-    [gr.outputs.Image(type="pil", label="Output"),gr.outputs.Image(type="pil", label="Output")],
+    [gr.outputs.Image(type="pil", label="Output")],
     title=title,
     description=description,
     article=article,