fix project layout

rembg/cmd/cli.py → rembg/cli.py

@@ -6,7 +6,7 @@ from distutils.util import strtobool
 import filetype
 from tqdm import tqdm
 
-from ..bg import remove
+from .bg import remove
 
 
 def main():

rembg/cmd/server.py → rembg/server.py

@@ -8,7 +8,7 @@ from urllib.request import urlopen
 from flask import Flask, request, send_file
 from waitress import serve
 
-from ..bg import remove
+from .bg import remove
 
 app = Flask(__name__)
 

setup.py

@@ -27,13 +27,13 @@ setup(
         "Programming Language :: Python :: 3 :: Only",
     ],
     keywords="remove, background, u2net",
-    packages=find_packages(include=["rembg"]),
+    packages=["rembg"],
     python_requires=">=3.8, <4",
     install_requires=requireds,
     entry_points={
         "console_scripts": [
-            "rembg=rembg.cmd.cli:main",
-            "rembg-server=rembg.cmd.server:main",
+            "rembg=rembg.cli:main",
+            "rembg-server=rembg.server:main",
         ],
     },
     version=versioneer.get_version(),