init

app.py

@@ -134,7 +134,7 @@ def process(im):
 
     print(results)
 
-    return Image.open(results[0])
+    return Image.open(results[0]), Image.open(im.name)
         
 title = "U-2-Net"
 description = "Gradio demo for U-2-Net, https://github.com/xuebinqin/U-2-Net"
@@ -144,7 +144,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,

modnet.py

@@ -60,7 +60,6 @@ class ModNet:
         im_h, im_w, im_c = im.shape
         x, y = self.get_scale_factor(im_h, im_w, ref_size)
 
-        image = im
         # resize image
         im = cv2.resize(im, None, fx=x, fy=y, interpolation=cv2.INTER_AREA)
 
@@ -78,7 +77,9 @@ class ModNet:
         matte = cv2.resize(matte, dsize=(im_w, im_h), interpolation=cv2.INTER_AREA)
 
         # obtain predicted foreground
-        image = np.asarray(image)
+        image = cv2.imread(image_path)
+        image = cv2.cvtColor(image, cv2.COLOR_BGR2RGB)
+
         if len(image.shape) == 2:
             image = image[:, :, None]
         if image.shape[2] == 1: