Update app.py

app.py

@@ -15,7 +15,9 @@ from rembg import remove
 def remove_background(img):
     img = Image.open(img)
     img = rembg.remove(img)
-    return img
+    output = remove(input_img[:, :, [2,1,0]])
+    return output
+
 
 def inference(img):
    output = remove_background(img)
@@ -27,6 +29,11 @@ def inference(img):
    cv2.drawContours(silhouette, contours, -1, (255, 255, 255), 1)
    return silhouette
 
+
+
+
+    
+
 gr.Interface(
    inference,
    gr.inputs.Image(type="filepath", label="Input"),