Minimal code implementation

README.md

@@ -1,13 +1,14 @@
 ---
 title: Rembg
-emoji: 🦀
-colorFrom: pink
+emoji: 👀
+colorFrom: yellow
 colorTo: yellow
 sdk: gradio
-sdk_version: 5.1.0
+sdk_version: 5.46.1
 app_file: app.py
 pinned: false
 license: mit
+short_description: 'Remove images background tool'
 ---
 
-Check out the configuration reference at https://huggingface.co/docs/hub/spaces-config-reference
+Check out the configuration reference at https://huggingface.co/docs/hub/spaces-config-reference

app.py

@@ -1,132 +1,87 @@
-import logging
+#######################################################################################
+#
+# MIT License
+#
+# Copyright (c) [2025] [[email protected]]
+#
+# Permission is hereby granted, free of charge, to any person obtaining a copy
+# of this software and associated documentation files (the "Software"), to deal
+# in the Software without restriction, including without limitation the rights
+# to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
+# copies of the Software, and to permit persons to whom the Software is
+# furnished to do so, subject to the following conditions:
+#
+# The above copyright notice and this permission notice shall be included in all
+# copies or substantial portions of the Software.
+#
+# THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
+# IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
+# FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
+# AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
+# LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
+# OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
+# SOFTWARE.
+#
+#######################################################################################
+
+# Implements an API endpoint for background image removal.
+#
+# This project is one of several repositories exploring image segmentation techniques.
+# All related projects and interactive demos can be found at:
+# https://huggingface.co/spaces/leonelhs/removators
+# Self app: https://huggingface.co/spaces/leonelhs/rembg
+#
+# Source code is based on or inspired by several projects.
+# For more details and proper attribution, please refer to the following resources:
+#
+# - [Rembg] - [https://github.com/danielgatis/rembg]
+# - [huggingface] [https://huggingface.co/spaces/KenjieDec/RemBG]
+#
 
 import gradio as gr
+import numpy as np
+from PIL import Image
 from rembg import new_session
+from rembg.bg import post_process
 
-from cutter import remove, make_label
-from utils import *
-
-remove_bg_models = {
-    "U2NET": "u2net",
-    "U2NET Human Seg": "u2net_human_seg",
-    "U2NET Cloth Seg": "u2net_cloth_seg"
+MODELS = {
+    "General segmentation": "u2net",
+    "Human segmentation": "u2net_human_seg",
+    "Cloth segmentation": "u2net_cloth_seg"
 }
 
-model_choices = keys(remove_bg_models)
-
-
-def predict(image, session, smoot, matting, bg_color):
-
-    session = new_session(remove_bg_models[session])
-
-    try:
-        return remove(session, image, smoot, matting, bg_color)
-    except ValueError as err:
-        logging.error(err)
-        return make_label(str(err)), None
-
-
-def change_show_mask(chk_state):
-    return gr.Image(visible=chk_state)
-
-
-def change_include_matting(chk_state):
-    return gr.Group(visible=chk_state), (0, 0, 0), 0, 0, 0
-
-
-def change_foreground_threshold(fg_value, value):
-    fg, bg, erode = value
-    return fg_value, bg, erode
-
-
-def change_background_threshold(bg_value, value):
-    fg, bg, erode = value
-    return fg, bg_value, erode
-
-
-def change_erode_size(erode_value, value):
-    fg, bg, erode = value
-    return fg, bg, erode_value
-
-
-def set_dominant_color(chk_state):
-    return chk_state, gr.ColorPicker(visible=not chk_state)
-
-
-def change_picker_color(picker, dominant):
-    if not dominant:
-        return picker
-    return dominant
-
-
-def change_background_mode(chk_state):
-    return gr.ColorPicker(visible=chk_state), \
-        gr.Checkbox(value=False, visible=chk_state)
-
-
-footer = r"""
-<center>
-<b>
-Demo based on <a href='https://github.com/danielgatis/rembg'>Rembg</a>
-</b>
-</center>
-"""
-
-with gr.Blocks(title="Remove background") as app:
-    color_state = gr.State(value=False)
-    matting_state = gr.State(value=(0, 0, 0))
-
-    gr.HTML("<center><h1>Remove Background Tool</h1></center>")
-    with gr.Row(equal_height=False):
-        with gr.Column():
-            input_img = gr.Image(type="pil", label="Input image")
-            drp_models = gr.Dropdown(choices=model_choices, label="Model Segment", value="U2NET")
+def predict(image, session="u2net"):
+    """
+        Remove the background from an image.
+        The function extracts the foreground and generates both a background-removed
+        image and a binary mask.
+        Parameters:
+            image (pil): File path to the input image.
+            session (string): Model for generate cutting mask.
+        Returns:
+            paths (tuple): paths for background-removed image and cutting mask.
+    """
+    session = new_session(session)
+    mask = session.predict(image)[0]
+    smoot_mask = Image.fromarray(post_process(np.array(mask)))
+    image.putalpha(smoot_mask)
+    return image, smoot_mask
+
+with gr.Blocks(title="Rembg") as app:
+    gr.Markdown("## Remove Background Tool")
+    with gr.Row():
+        with gr.Column(scale=1):
+            inp = gr.Image(type="pil", label="Upload Image")
+            sess = gr.Dropdown(choices=list(MODELS.items()), label="Model Segment", value="u2net")
+            btn_predict = gr.Button("Remove background")
+        with gr.Column(scale=2):
             with gr.Row():
-                chk_include_matting = gr.Checkbox(label="Matting", value=False)
-                chk_smoot_mask = gr.Checkbox(label="Smoot Mask", value=False)
-                chk_show_mask = gr.Checkbox(label="Show Mask", value=False)
-            with gr.Group(visible=False) as slider_matting:
-                slr_fg_threshold = gr.Slider(0, 300, value=270, step=1, label="Alpha matting foreground threshold")
-                slr_bg_threshold = gr.Slider(0, 50, value=20, step=1, label="Alpha matting background threshold")
-                slr_erode_size = gr.Slider(0, 20, value=11, step=1, label="Alpha matting erode size")
-            with gr.Group():
-                with gr.Row():
-                    chk_change_color = gr.Checkbox(label="Change background color", value=False)
-                    pkr_color = gr.ColorPicker(label="Pick a new color", visible=False)
-                    chk_dominant = gr.Checkbox(label="Use dominant color", value=False, visible=False)
-            run_btn = gr.Button(value="Remove background", variant="primary")
-        with gr.Column():
-            output_img = gr.Image(type="pil", label="Image Result")
-            mask_img = gr.Image(type="pil", label="Image Mask", visible=False)
-            gr.ClearButton(components=[input_img, output_img, mask_img])
-
-    chk_include_matting.change(change_include_matting, inputs=[chk_include_matting],
-                               outputs=[slider_matting, matting_state,
-                                        slr_fg_threshold, slr_bg_threshold, slr_erode_size])
+                with gr.Column(scale=1):
+                    out = gr.Image(type="pil", label="Output image")
+                    with gr.Accordion("See intermediates", open=False):
+                        out_mask = gr.Image(type="pil", label="Mask")
 
-    slr_fg_threshold.change(change_foreground_threshold, inputs=[slr_fg_threshold, matting_state],
-                            outputs=[matting_state])
-
-    slr_bg_threshold.change(change_background_threshold, inputs=[slr_bg_threshold, matting_state],
-                            outputs=[matting_state])
-
-    slr_erode_size.change(change_erode_size, inputs=[slr_erode_size, matting_state],
-                          outputs=[matting_state])
-
-    chk_show_mask.change(change_show_mask, inputs=[chk_show_mask], outputs=[mask_img])
-
-    chk_change_color.change(change_background_mode, inputs=[chk_change_color],
-                            outputs=[pkr_color, chk_dominant])
-
-    pkr_color.change(change_picker_color, inputs=[pkr_color, chk_dominant], outputs=[color_state])
-
-    chk_dominant.change(set_dominant_color, inputs=[chk_dominant], outputs=[color_state, pkr_color])
-
-    run_btn.click(predict, inputs=[input_img, drp_models, chk_smoot_mask, matting_state, color_state],
-                  outputs=[output_img, mask_img])
-
-    with gr.Row():
-        gr.HTML(footer)
+    btn_predict.click(predict, inputs=[inp, sess], outputs=[out, out_mask])
 
-app.queue()
-app.launch(share=False, debug=True, show_error=True)
+app.launch(share=False, debug=True, show_error=True, mcp_server=True, pwa=True)
+app.queue()

notes.txt

@@ -0,0 +1,3 @@
+https://github.com/danielgatis/rembg/releases/download/v0.0.0/u2net.onnx
+https://github.com/danielgatis/rembg/releases/download/v0.0.0/u2net_human_seg.onnx
+https://github.com/danielgatis/rembg/releases/download/v0.0.0/u2net_cloth_seg.onnx

old_app.py

@@ -0,0 +1,132 @@
+import logging
+
+import gradio as gr
+from rembg import new_session
+
+from cutter import remove, make_label
+from utils import *
+
+remove_bg_models = {
+    "U2NET": "u2net",
+    "U2NET Human Seg": "u2net_human_seg",
+    "U2NET Cloth Seg": "u2net_cloth_seg"
+}
+
+model_choices = keys(remove_bg_models)
+
+
+def predict(image, session, smoot, matting, bg_color):
+
+    session = new_session(remove_bg_models[session])
+
+    try:
+        return remove(session, image, smoot, matting, bg_color)
+    except ValueError as err:
+        logging.error(err)
+        return make_label(str(err)), None
+
+
+def change_show_mask(chk_state):
+    return gr.Image(visible=chk_state)
+
+
+def change_include_matting(chk_state):
+    return gr.Group(visible=chk_state), (0, 0, 0), 0, 0, 0
+
+
+def change_foreground_threshold(fg_value, value):
+    fg, bg, erode = value
+    return fg_value, bg, erode
+
+
+def change_background_threshold(bg_value, value):
+    fg, bg, erode = value
+    return fg, bg_value, erode
+
+
+def change_erode_size(erode_value, value):
+    fg, bg, erode = value
+    return fg, bg, erode_value
+
+
+def set_dominant_color(chk_state):
+    return chk_state, gr.ColorPicker(visible=not chk_state)
+
+
+def change_picker_color(picker, dominant):
+    if not dominant:
+        return picker
+    return dominant
+
+
+def change_background_mode(chk_state):
+    return gr.ColorPicker(visible=chk_state), \
+        gr.Checkbox(value=False, visible=chk_state)
+
+
+footer = r"""
+<center>
+<b>
+Demo based on <a href='https://github.com/danielgatis/rembg'>Rembg</a>
+</b>
+</center>
+"""
+
+with gr.Blocks(title="Remove background") as app:
+    color_state = gr.State(value=False)
+    matting_state = gr.State(value=(0, 0, 0))
+
+    gr.HTML("<center><h1>Remove Background Tool</h1></center>")
+    with gr.Row(equal_height=False):
+        with gr.Column():
+            input_img = gr.Image(type="pil", label="Input image")
+            drp_models = gr.Dropdown(choices=model_choices, label="Model Segment", value="U2NET")
+            with gr.Row():
+                chk_include_matting = gr.Checkbox(label="Matting", value=False)
+                chk_smoot_mask = gr.Checkbox(label="Smoot Mask", value=False)
+                chk_show_mask = gr.Checkbox(label="Show Mask", value=False)
+            with gr.Group(visible=False) as slider_matting:
+                slr_fg_threshold = gr.Slider(0, 300, value=270, step=1, label="Alpha matting foreground threshold")
+                slr_bg_threshold = gr.Slider(0, 50, value=20, step=1, label="Alpha matting background threshold")
+                slr_erode_size = gr.Slider(0, 20, value=11, step=1, label="Alpha matting erode size")
+            with gr.Group():
+                with gr.Row():
+                    chk_change_color = gr.Checkbox(label="Change background color", value=False)
+                    pkr_color = gr.ColorPicker(label="Pick a new color", visible=False)
+                    chk_dominant = gr.Checkbox(label="Use dominant color", value=False, visible=False)
+            run_btn = gr.Button(value="Remove background", variant="primary")
+        with gr.Column():
+            output_img = gr.Image(type="pil", label="Image Result")
+            mask_img = gr.Image(type="pil", label="Image Mask", visible=False)
+            gr.ClearButton(components=[input_img, output_img, mask_img])
+
+    chk_include_matting.change(change_include_matting, inputs=[chk_include_matting],
+                               outputs=[slider_matting, matting_state,
+                                        slr_fg_threshold, slr_bg_threshold, slr_erode_size])
+
+    slr_fg_threshold.change(change_foreground_threshold, inputs=[slr_fg_threshold, matting_state],
+                            outputs=[matting_state])
+
+    slr_bg_threshold.change(change_background_threshold, inputs=[slr_bg_threshold, matting_state],
+                            outputs=[matting_state])
+
+    slr_erode_size.change(change_erode_size, inputs=[slr_erode_size, matting_state],
+                          outputs=[matting_state])
+
+    chk_show_mask.change(change_show_mask, inputs=[chk_show_mask], outputs=[mask_img])
+
+    chk_change_color.change(change_background_mode, inputs=[chk_change_color],
+                            outputs=[pkr_color, chk_dominant])
+
+    pkr_color.change(change_picker_color, inputs=[pkr_color, chk_dominant], outputs=[color_state])
+
+    chk_dominant.change(set_dominant_color, inputs=[chk_dominant], outputs=[color_state, pkr_color])
+
+    run_btn.click(predict, inputs=[input_img, drp_models, chk_smoot_mask, matting_state, color_state],
+                  outputs=[output_img, mask_img])
+
+    with gr.Row():
+        gr.HTML(footer)
+
+app.queue()
+app.launch(share=False, debug=True, show_error=True)

requirements.txt

@@ -1,7 +1,2 @@
-rembg~=2.0.47
-pillow~=10.4.0
-pymatting~=1.1.12
-opencv-python-headless
-gradio~=5.1.0
-numpy~=1.26.4
-scipy~=1.14.1
+rembg>=2.0.67
+onnxruntime>=1.22.1