update stuff

.github/FUNDING.yml

@@ -1,2 +1,2 @@
-# github: [danielgatis]
-# custom: ["https://www.buymeacoffee.com/danielgatis"]
+ github: [rnag]
+ custom: ["https://www.buymeacoffee.com/ritviknag"]

.github/workflows/test-install.yml

@@ -19,6 +19,7 @@ jobs:
               run: |
                   python -m pip install --upgrade pip
                   pip install .
+                  pip install -r requirements-dev.txt
                   pip install pytest
             - name: Test installation with pytest
               run: |

.github/workflows/test-specific-environment.yml

@@ -20,6 +20,7 @@ jobs:
                   python -m pip install --upgrade pip
                   pip install pytest
                   pip install -r requirements.txt
+                  pip install -r requirements-dev.txt
             - name: Test with pytest
               run: |
                   PYTHONPATH=$PYTHONPATH:. pytest .

rembg/bg.py

@@ -13,6 +13,7 @@
 )
 from PIL import Image
 from PIL.Image import Image as PILImage
+
 # from pymatting.alpha.estimate_alpha_cf import estimate_alpha_cf
 # from pymatting.foreground.estimate_foreground_ml import estimate_foreground_ml
 # from pymatting.util.util import stack_images
@@ -116,7 +117,6 @@ def remove(
     only_mask: bool = False,
     post_process_mask: bool = False,
 ) -> Union[bytes, PILImage, np.ndarray]:
-
     if isinstance(data, PILImage):
         return_type = ReturnType.PILLOW
         img = data

rembg/session_factory.py

@@ -5,6 +5,7 @@
 import onnxruntime as ort
 
 from .session_base import BaseSession
+
 # from .session_cloth import ClothSession
 from .session_simple import SimpleSession
 

requirements-dev.txt

@@ -0,0 +1 @@
+imagehash==4.3.1

requirements.txt

@@ -1,18 +1,4 @@
-# aiohttp==3.8.1
-# asyncer==0.0.2
-# click==8.1.3
-# fastapi==0.87.0
-# filetype==1.2.0
-# pooch==1.6.0
-imagehash==4.3.1
 numpy==1.23.5
 onnxruntime==1.13.1
 opencv-python-headless==4.6.0.66
 pillow==9.3.0
-# pymatting==1.1.8
-# python-multipart==0.0.5
-# scikit-image==0.19.3
-# scipy==1.9.3
-# tqdm==4.64.1
-# uvicorn==0.20.0
-# watchdog==2.1.9

setup.py

@@ -3,7 +3,7 @@
 import sys
 
 sys.path.append(os.path.dirname(__file__))
-from setuptools import find_packages, setup
+from setuptools import setup
 
 import versioneer
 
@@ -16,7 +16,7 @@
     description="Remove image background",
     long_description=long_description,
     long_description_content_type="text/markdown",
-    url="https://github.com/rnag/rembg",
+    url="https://github.com/rnag/rembg-aws-lambda",
     author="Daniel Gatis",
     author_email="[email protected]",
     classifiers=[
@@ -39,30 +39,11 @@
     include_package_data=True,
     python_requires=">3.7, <3.11",
     install_requires=[
-        # "aiohttp~=3.8.1",
-        # "asyncer~=0.0.2",
-        # "click~=8.1.3",
-        # "fastapi~=0.87.0",
-        # "filetype~=1.2.0",
-        # "pooch~=1.6.0",
-        # "imagehash~=4.3.1",
         "numpy~=1.23.5",
         "onnxruntime~=1.13.1",
         "opencv-python-headless~=4.6.0.66",
         "pillow~=9.3.0",
-        # "pymatting~=1.1.8",
-        # "python-multipart~=0.0.5",
-        # "scikit-image~=0.19.3",
-        # "scipy~=1.9.3",
-        # "tqdm~=4.64.1",
-        # "uvicorn~=0.20.0",
-        # "watchdog~=2.1.9",
     ],
-    # entry_points={
-    #     "console_scripts": [
-    #         "rembg=rembg.cli:main",
-    #     ],
-    # },
     extras_require={
         "gpu": ["onnxruntime-gpu~=1.13.1"],
     },