Update app.py

app.py

@@ -25,17 +25,19 @@ def extract_object(image):
     pred = preds[0].squeeze()
     pred_pil = transforms.ToPILImage()(pred)
     mask = pred_pil.resize(image.size)
-    image_with_alpha = image.convert("RGBA")
-    image_with_alpha.putalpha(mask)
-    return image_with_alpha
+
+    # Create a new transparent image
+    result = Image.new("RGBA", image.size, (0, 0, 0, 0))
+    result.paste(image, mask=mask)
+    return result
 
 iface = gr.Interface(
     fn=extract_object,
     inputs=gr.Image(type="pil", label="Upload Image"),
-    outputs=gr.Image(type="pil", label="Segmented Image"),
+    outputs=gr.Image(type="pil", label="Object with Transparent Background"),
     title="BiRefNet Background Removal",
-    description="Upload an image and get the foreground object extracted."
+    description="Upload an image and get the foreground object extracted onto a transparent background."
 )
 
 if __name__ == "__main__":
-    iface.launch()
+    iface.launch()