diff --git a/setup.py b/setup.py index 2908dde..b6aa6e3 100644 --- a/setup.py +++ b/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(),