Update app.py

app.py

@@ -11,6 +11,7 @@ import cv2
 import gradio as gr
 import os
 from PIL import Image
+from rembg import remove 
 import numpy as np
 import torch
 from torch.autograd import Variable
@@ -138,11 +139,13 @@ def inference(image):
 
   pil_mask = Image.fromarray(mask).convert('L')
   im_rgb = Image.open(image).convert("RGB")
+  imrgba1 = img_rgb.copy()
+  imrgba2 = remove(imrgba1,210)
 
   im_rgba = im_rgb.copy()
   im_rgba.putalpha(pil_mask)
 
-  return im_rgba
+  return im_rgba, imrgba2
 
 
 title = "Bg remover for sarvm catalog"
@@ -152,7 +155,7 @@ article = "<div><center><img src='https://visitor-badge.glitch.me/badge?page_id=
 interface = gr.Interface(
     fn=inference,
     inputs=gr.Image(type='filepath'),
-    outputs="image",
+    outputs=["image","image"]
     title=title,
     description=description,
     article=article,