Upload 6 files

.gitattributes

@@ -1,35 +1,36 @@
-*.7z filter=lfs diff=lfs merge=lfs -text
-*.arrow filter=lfs diff=lfs merge=lfs -text
-*.bin filter=lfs diff=lfs merge=lfs -text
-*.bz2 filter=lfs diff=lfs merge=lfs -text
-*.ckpt filter=lfs diff=lfs merge=lfs -text
-*.ftz filter=lfs diff=lfs merge=lfs -text
-*.gz filter=lfs diff=lfs merge=lfs -text
-*.h5 filter=lfs diff=lfs merge=lfs -text
-*.joblib filter=lfs diff=lfs merge=lfs -text
-*.lfs.* filter=lfs diff=lfs merge=lfs -text
-*.mlmodel filter=lfs diff=lfs merge=lfs -text
-*.model filter=lfs diff=lfs merge=lfs -text
-*.msgpack filter=lfs diff=lfs merge=lfs -text
-*.npy filter=lfs diff=lfs merge=lfs -text
-*.npz filter=lfs diff=lfs merge=lfs -text
-*.onnx filter=lfs diff=lfs merge=lfs -text
-*.ot filter=lfs diff=lfs merge=lfs -text
-*.parquet filter=lfs diff=lfs merge=lfs -text
-*.pb filter=lfs diff=lfs merge=lfs -text
-*.pickle filter=lfs diff=lfs merge=lfs -text
-*.pkl filter=lfs diff=lfs merge=lfs -text
-*.pt filter=lfs diff=lfs merge=lfs -text
-*.pth filter=lfs diff=lfs merge=lfs -text
-*.rar filter=lfs diff=lfs merge=lfs -text
-*.safetensors filter=lfs diff=lfs merge=lfs -text
-saved_model/**/* filter=lfs diff=lfs merge=lfs -text
-*.tar.* filter=lfs diff=lfs merge=lfs -text
-*.tar filter=lfs diff=lfs merge=lfs -text
-*.tflite filter=lfs diff=lfs merge=lfs -text
-*.tgz filter=lfs diff=lfs merge=lfs -text
-*.wasm filter=lfs diff=lfs merge=lfs -text
-*.xz filter=lfs diff=lfs merge=lfs -text
-*.zip filter=lfs diff=lfs merge=lfs -text
-*.zst filter=lfs diff=lfs merge=lfs -text
-*tfevents* filter=lfs diff=lfs merge=lfs -text
+*.7z filter=lfs diff=lfs merge=lfs -text
+*.arrow filter=lfs diff=lfs merge=lfs -text
+*.bin filter=lfs diff=lfs merge=lfs -text
+*.bz2 filter=lfs diff=lfs merge=lfs -text
+*.ckpt filter=lfs diff=lfs merge=lfs -text
+*.ftz filter=lfs diff=lfs merge=lfs -text
+*.gz filter=lfs diff=lfs merge=lfs -text
+*.h5 filter=lfs diff=lfs merge=lfs -text
+*.joblib filter=lfs diff=lfs merge=lfs -text
+*.lfs.* filter=lfs diff=lfs merge=lfs -text
+*.mlmodel filter=lfs diff=lfs merge=lfs -text
+*.model filter=lfs diff=lfs merge=lfs -text
+*.msgpack filter=lfs diff=lfs merge=lfs -text
+*.npy filter=lfs diff=lfs merge=lfs -text
+*.npz filter=lfs diff=lfs merge=lfs -text
+*.onnx filter=lfs diff=lfs merge=lfs -text
+*.ot filter=lfs diff=lfs merge=lfs -text
+*.parquet filter=lfs diff=lfs merge=lfs -text
+*.pb filter=lfs diff=lfs merge=lfs -text
+*.pickle filter=lfs diff=lfs merge=lfs -text
+*.pkl filter=lfs diff=lfs merge=lfs -text
+*.pt filter=lfs diff=lfs merge=lfs -text
+*.pth filter=lfs diff=lfs merge=lfs -text
+*.rar filter=lfs diff=lfs merge=lfs -text
+*.safetensors filter=lfs diff=lfs merge=lfs -text
+saved_model/**/* filter=lfs diff=lfs merge=lfs -text
+*.tar.* filter=lfs diff=lfs merge=lfs -text
+*.tar filter=lfs diff=lfs merge=lfs -text
+*.tflite filter=lfs diff=lfs merge=lfs -text
+*.tgz filter=lfs diff=lfs merge=lfs -text
+*.wasm filter=lfs diff=lfs merge=lfs -text
+*.xz filter=lfs diff=lfs merge=lfs -text
+*.zip filter=lfs diff=lfs merge=lfs -text
+*.zst filter=lfs diff=lfs merge=lfs -text
+*tfevents* filter=lfs diff=lfs merge=lfs -text
+examples/butterfly.jpg filter=lfs diff=lfs merge=lfs -text

.gitignore

@@ -0,0 +1,2 @@
+__pycache__/
+*.pyc

README.md

@@ -1,13 +1,24 @@
----
-title: BackgraoundRemover
-emoji: 🌍
-colorFrom: gray
-colorTo: purple
-sdk: gradio
-sdk_version: 6.20.0
-python_version: '3.12'
-app_file: app.py
-pinned: false
----
-
-Check out the configuration reference at https://huggingface.co/docs/hub/spaces-config-reference
+---
+title: Background Remover
+emoji: 🪄
+colorFrom: gray
+colorTo: purple
+sdk: gradio
+sdk_version: 6.20.0
+python_version: '3.12'
+app_file: app.py
+pinned: false
+---
+
+# 🪄 Background Remover
+
+Remove image backgrounds instantly using the open-source [BiRefNet](https://huggingface.co/ZhengPeng7/BiRefNet)
+segmentation model. Free to use, no sign-up, no watermark — runs on Hugging Face's free
+[ZeroGPU](https://huggingface.co/docs/hub/spaces-zerogpu).
+
+Three ways to use it:
+- **Upload** — drag and drop an image, get a transparent PNG back.
+- **Image URL** — paste a link to any image on the web.
+- **File → PNG** — upload a file and download a transparent PNG (also exposed as the `png` API for automation / MCP).
+
+Check out the configuration reference at https://huggingface.co/docs/hub/spaces-config-reference

app.py

@@ -0,0 +1,160 @@
+import os
+from typing import Tuple, Union
+
+import gradio as gr
+import spaces
+import torch
+from loadimg import load_img
+from PIL import Image
+from torchvision import transforms
+from transformers import AutoModelForImageSegmentation
+
+torch.set_float32_matmul_precision("high")
+
+DEVICE = "cuda" if torch.cuda.is_available() else "cpu"
+
+birefnet = AutoModelForImageSegmentation.from_pretrained(
+    "ZhengPeng7/BiRefNet", trust_remote_code=True
+)
+birefnet.to(DEVICE)
+birefnet.eval()
+
+# Keep inference inputs aligned with the loaded model precision (fp16/fp32).
+MODEL_DTYPE = next(birefnet.parameters()).dtype
+
+transform_image = transforms.Compose(
+    [
+        transforms.Resize((1024, 1024)),
+        transforms.ToTensor(),
+        transforms.Normalize([0.485, 0.456, 0.406], [0.229, 0.224, 0.225]),
+    ]
+)
+
+EXAMPLE_DIR = os.path.join(os.path.dirname(os.path.abspath(__file__)), "examples")
+EXAMPLE_IMAGE = os.path.join(EXAMPLE_DIR, "butterfly.jpg")
+EXAMPLE_URL = (
+    "https://hips.hearstapps.com/hmg-prod/images/gettyimages-1229892983-square.jpg"
+)
+
+
+@spaces.GPU
+def process(image: Image.Image) -> Image.Image:
+    """
+    Apply BiRefNet-based image segmentation to remove the background.
+    Args:
+        image (PIL.Image): The input RGB image.
+    Returns:
+        PIL.Image: The image with the background removed, using the segmentation mask as transparency.
+    """
+    image_size = image.size
+    input_images = (
+        transform_image(image).unsqueeze(0).to(device=DEVICE, dtype=MODEL_DTYPE)
+    )
+    with torch.no_grad():
+        preds = birefnet(input_images)[-1].sigmoid().cpu()
+    pred = preds[0].squeeze()
+    pred_pil = transforms.ToPILImage()(pred)
+    mask = pred_pil.resize(image_size)
+    image.putalpha(mask)
+    return image
+
+
+def remove_bg(image: Union[Image.Image, str]) -> Tuple[Image.Image, Image.Image]:
+    """Remove the background from an uploaded image or an image URL."""
+    im = load_img(image, output_type="pil").convert("RGB")
+    origin = im.copy()
+    return origin, process(im)
+
+
+def remove_bg_to_file(f: str) -> str:
+    """Load an image file from disk, remove the background, and save as a transparent PNG."""
+    name_path = os.path.splitext(f)[0] + ".png"
+    im = load_img(f, output_type="pil").convert("RGB")
+    process(im).save(name_path)
+    return name_path
+
+
+THEME = gr.themes.Soft(
+    primary_hue="violet",
+    secondary_hue="purple",
+    neutral_hue="slate",
+    font=[gr.themes.GoogleFont("Inter"), "ui-sans-serif", "system-ui", "sans-serif"],
+)
+
+CSS = """
+.gradio-container { max-width: 980px !important; margin: 0 auto !important; }
+#hero { text-align: center; padding: 6px 0 2px; }
+#hero h1 { font-size: 2.15rem; margin-bottom: 2px; }
+#hero p { color: var(--body-text-color-subdued); font-size: 1.02rem; margin-top: 0; }
+#run-btn { min-height: 46px; font-weight: 600; }
+.footer-note { text-align: center; opacity: 0.65; font-size: 0.85rem; margin-top: 6px; }
+.footer-note a { color: inherit; }
+"""
+
+with gr.Blocks(title="Background Remover") as demo:
+    gr.Markdown(
+        """
+        <div id="hero">
+        <h1>🪄 Background Remover</h1>
+        <p>Drop an image, get a clean transparent cutout. Powered by the open-source
+        <a href="https://huggingface.co/ZhengPeng7/BiRefNet" target="_blank">BiRefNet</a> model — free, no sign-up, no watermark.</p>
+        </div>
+        """
+    )
+
+    with gr.Tabs():
+        with gr.Tab("📤 Upload"):
+            with gr.Row():
+                with gr.Column():
+                    upload_in = gr.Image(
+                        label="Source image", type="pil", sources=["upload", "clipboard"]
+                    )
+                    upload_btn = gr.Button(
+                        "Remove background", variant="primary", elem_id="run-btn"
+                    )
+                with gr.Column():
+                    upload_out = gr.ImageSlider(
+                        label="Result — drag to compare", type="pil", format="png"
+                    )
+            gr.Examples(examples=[EXAMPLE_IMAGE], inputs=upload_in, label="Try an example")
+            upload_btn.click(remove_bg, inputs=upload_in, outputs=upload_out, api_name="image")
+
+        with gr.Tab("🔗 Image URL"):
+            with gr.Row():
+                with gr.Column():
+                    url_in = gr.Textbox(
+                        label="Image URL", placeholder="https://example.com/photo.jpg"
+                    )
+                    url_btn = gr.Button(
+                        "Remove background", variant="primary", elem_id="run-btn"
+                    )
+                with gr.Column():
+                    url_out = gr.ImageSlider(
+                        label="Result — drag to compare", type="pil", format="png"
+                    )
+            gr.Examples(examples=[EXAMPLE_URL], inputs=url_in, label="Try an example")
+            url_btn.click(remove_bg, inputs=url_in, outputs=url_out, api_name="text")
+
+        with gr.Tab("📁 File → PNG"):
+            gr.Markdown("Upload a file and get back a downloadable transparent PNG — handy for automation, the API, or MCP.")
+            with gr.Row():
+                file_in = gr.Image(label="Source image", type="filepath")
+                file_out = gr.File(label="Transparent PNG")
+            file_btn = gr.Button("Generate PNG", variant="primary", elem_id="run-btn")
+            gr.Examples(examples=[EXAMPLE_IMAGE], inputs=file_in, label="Try an example")
+            file_btn.click(remove_bg_to_file, inputs=file_in, outputs=file_out, api_name="png")
+
+    gr.Markdown(
+        """
+        <div class="footer-note">
+        Model: <a href="https://huggingface.co/ZhengPeng7/BiRefNet" target="_blank">ZhengPeng7/BiRefNet</a> ·
+        UI: <a href="https://gradio.app" target="_blank">Gradio</a> ·
+        Runs free on <a href="https://huggingface.co/docs/hub/spaces-zerogpu" target="_blank">ZeroGPU</a><br/>
+        Prefer a dedicated API? Use <a href="https://huggingface.co/spaces/hf-applications/background-removal" target="_blank">hf-applications/background-removal</a>,
+        or call the <code>png</code> endpoint from the "File → PNG" tab above.
+        </div>
+        """
+    )
+
+if __name__ == "__main__":
+    demo.launch(show_error=True, mcp_server=True, theme=THEME, css=CSS)

requirements.txt

@@ -0,0 +1,16 @@
+gradio
+torch
+torchvision
+transformers>=4.39.1
+accelerate
+spaces
+loadimg>=0.1.1
+pillow
+numpy
+opencv-python
+timm
+kornia
+einops
+prettytable
+scikit-image
+huggingface_hub