Update app.py

app.py

@@ -14,32 +14,18 @@ import rembg
 from rembg import remove
 
 
-def remove_background(img):
+def inference(img):
     img = Image.open(img)
     img = np.array(img)
     input_img = cv2.imread(img)
-    if input_img is None:
-        raise Exception("La imagen no se ha podido cargar correctamente.")
     output = remove(input_img[:, :, [2,1,0]])
-    return output
-
-
-def silhouette(img):
-    input_img = cv2.imdecode(np.frombuffer(img, np.uint8), cv2.IMREAD_COLOR)
-    if input_img is None:
-        raise Exception("La imagen no se ha podido cargar correctamente.")
-    gray = cv2.cvtColor(input_img, cv2.COLOR_BGR2GRAY)
+    gray = cv2.cvtColor(output, cv2.COLOR_BGR2GRAY)
     edges = cv2.Canny(gray, 50, 150)
     contours, _ = cv2.findContours(edges, cv2.RETR_EXTERNAL, cv2.CHAIN_APPROX_SIMPLE)
-    silhouette = np.zeros_like(input_img)
+    silhouette = np.zeros_like(output)
     cv2.drawContours(silhouette, contours, -1, (255, 255, 255), 1)
     return silhouette
 
-def inference(img):
-    img = remove_background(img)
-    img = silhouette(img)
-    return img
-
 # Crear interfaz de usuario con Gradio
 gr.Interface(
     inference,