minimal code

README.md

@@ -1,14 +1,13 @@
 ---
 title: Rembg
-emoji: 👀
-colorFrom: yellow
+emoji: 🦀
+colorFrom: pink
 colorTo: yellow
 sdk: gradio
-sdk_version: 5.46.1
+sdk_version: 5.1.0
 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,96 +1,132 @@
-#######################################################################################
-#
-# 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/removatorsau
-# 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 logging
 
 import gradio as gr
-import numpy as np
-from PIL import Image
 from rembg import new_session
-from rembg.bg import post_process
 
-MODELS = {
-    "General segmentation": "u2net",
-    "Human segmentation": "u2net_human_seg",
-    "Cloth segmentation": "u2net_cloth_seg"
+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"
 }
 
-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
+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="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.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():
-                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")
+                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])
 
-    btn_predict.click(predict, inputs=[inp, sess], outputs=[out, out_mask])
+    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.launch(share=False, debug=True, show_error=True, mcp_server=True, pwa=True)
-app.queue()
+app.queue()
+app.launch(share=False, debug=True, show_error=True)

notes.txt

@@ -1,3 +0,0 @@
-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

@@ -1,132 +0,0 @@
-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,2 +1,7 @@
-rembg>=2.0.67
-onnxruntime>=1.22.1
+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