Update app.py

app.py

@@ -6,20 +6,19 @@ from PIL import Image
 # Load Hugging Face background removal model
 # -------------------------------
 pipe = pipeline(
-    "image-segmentation",
+    task="image-segmentation",
     model="briaai/RMBG-1.4",
-    trust_remote_code=True   # <-- IMPORTANT FIX
+    trust_remote_code=True
 )
 
 # -------------------------------
 # Function to remove background
 # -------------------------------
 def remove_bg(image: Image.Image):
-    # Run model to get mask
-    result = pipe(image)
-    mask = result[0]["mask"]  # segmentation mask (white=foreground, black=background)
+    # Run model → this returns a PIL mask directly
+    mask = pipe(image)
 
-    # Convert mask to binary
+    # Ensure mask is grayscale
     mask = mask.convert("L")
 
     # Apply mask to original image
@@ -37,7 +36,7 @@ demo = gr.Interface(
     inputs=gr.Image(type="pil", label="Upload an image"),
     outputs=gr.Image(type="pil", label="Background Removed"),
     title="Background Remover",
-    description="Upload an image to remove its background using Hugging Face model.",
+    description="Upload an image to remove its background using Hugging Face RMBG model.",
 )
 
 # -------------------------------