Update app.py

app.py

@@ -13,6 +13,10 @@ from io import BytesIO
 import rembg
 from rembg import remove
 
+def inference(img):
+   input_img = cv2.imread(img)
+   output = remove(input_img[:, :, [2,1,0]])
+   return output
 
 def inference(img):
     img = Image.open(img)