Merge pull request #358 from jangop/main

.github/workflows/test-install.yml

@@ -0,0 +1,25 @@
+name: Test installation
+
+on: [push]
+
+jobs:
+    build:
+        runs-on: ubuntu-latest
+        strategy:
+            matrix:
+                python-version: ["3.8", "3.9", "3.10"]
+
+        steps:
+            - uses: actions/checkout@v3
+            - name: Set up Python ${{ matrix.python-version }}
+              uses: actions/setup-python@v4
+              with:
+                  python-version: ${{ matrix.python-version }}
+            - name: Install package
+              run: |
+                  python -m pip install --upgrade pip
+                  pip install .
+                  pip install pytest
+            - name: Test installation with pytest
+              run: |
+                  pytest

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

@@ -1,4 +1,4 @@
-name: Run tests
+name: Test in specific environment
 
 on: [push]
 
@@ -15,7 +15,7 @@ jobs:
               uses: actions/setup-python@v4
               with:
                   python-version: ${{ matrix.python-version }}
-            - name: Install dependencies
+            - name: Set up environment
               run: |
                   python -m pip install --upgrade pip
                   pip install pytest

setup.py

@@ -11,12 +11,6 @@ here = pathlib.Path(__file__).parent.resolve()
 
 long_description = (here / "README.md").read_text(encoding="utf-8")
 
-with open(here / "requirements.txt") as f:
-    requireds = f.read().splitlines()
-
-with open(here / "requirements-gpu.txt") as f:
-    gpu_requireds = f.read().splitlines()
-
 setup(
     name="rembg",
     description="Remove image background",
@@ -42,14 +36,33 @@ setup(
     keywords="remove, background, u2net",
     packages=["rembg"],
     python_requires=">3.7, <3.11",
-    install_requires=requireds,
+    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": gpu_requireds,
+        "gpu": ["onnxruntime-gpu~=1.13.1"],
     },
     version=versioneer.get_version(),
     cmdclass=versioneer.get_cmdclass(),