diff --git a/.gitignore b/.gitignore index 73f6779..48cccc1 100644 --- a/.gitignore +++ b/.gitignore @@ -3,6 +3,7 @@ build/ dist/ .venv/ .direnv/ +*.spec *.egg-info/ *.egg *.py[cod] diff --git a/rembg.py b/rembg.py new file mode 100644 index 0000000..48c8f56 --- /dev/null +++ b/rembg.py @@ -0,0 +1,4 @@ +from rembg.cli import main + +if __name__ == '__main__': + main() diff --git a/rembg/cli.py b/rembg/cli.py index 0d46155..6e8c5bc 100644 --- a/rembg/cli.py +++ b/rembg/cli.py @@ -22,7 +22,7 @@ from .session_factory import new_session @click.group() -@click.version_option() +@click.version_option(version=_version.get_versions()["version"]) def main() -> None: pass @@ -351,7 +351,3 @@ def s(port: int, log_level: str) -> None: return await asyncify(im_without_bg)(file, commons) uvicorn.run(app, host="0.0.0.0", port=port, log_level=log_level) - - -if __name__ == "__main__": - main()