Update app.py

app.py

@@ -2,33 +2,19 @@ import os
 os.system('/usr/local/bin/python -m pip install --upgrade pip')
 os.system("pip install opencv-python")
 os.system("pip install pillow")
+os.system("pip install rembg")
 
 import gradio as gr
 import cv2
 import numpy as np
 from PIL import Image
 
-def remove_background(img):
-    # Load the image
-    img = cv2.imread(img)
-
-    # Define the initial mask
-    mask = np.zeros(img.shape[:2], dtype=np.uint8)
-    bgdModel = np.zeros((1, 65), np.float64)
-    fgdModel = np.zeros((1, 65), np.float64)
-
-    # Define the rectangle for the GrabCut algorithm
-    rect = (50, 50, img.shape[1]-100, img.shape[0]-100)
-
-    # Run the GrabCut algorithm
-    cv2.grabCut(img, mask, rect, bgdModel, fgdModel, 5, cv2.GC_INIT_WITH_RECT)
-
-    # Get the final mask
-    mask2 = np.where((mask==2)|(mask==0), 0, 1).astype('uint8')
-
-    # Multiply the image with the mask to remove the background
-    img = img * mask2[:, :, np.newaxis]
+import rembg
+from rembg import remove
 
+def remove_background(img):
+    img = Image.open(img)
+    img = rembg.remove(img)
     return img
 
 def inference(img):
@@ -48,4 +34,4 @@ gr.Interface(
    description="Ingresa una imagen para convertirla a silueta",
    article="",
    css="Footer {visibility: hidden}"
-).launch()
+).launch()