AI/rembg
1
0
Fork 0
mirror of https://git.mirrors.martin98.com/https://github.com/danielgatis/rembg synced 2025-08-17 10:05:54 +08:00
Code Issues Packages Projects Releases Wiki Activity

fix issue #2

Browse Source
This commit is contained in:
Daniel Gatis 2020-08-19 10:32:38 -03:00
parent 28100b3a52
commit 1478c5845c
2 changed files with 3 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
setup.py
View File

@ -11,7 +11,7 @@ with open("requirements.txt") as f:
setup( setup(
name="rembg", name="rembg",
version="1.0.5", version="1.0.6",
description="Remove image background", description="Remove image background",
long_description=long_description, long_description=long_description,
long_description_content_type="text/markdown", long_description_content_type="text/markdown",

2
src/rembg/u2net/detect.py
View File

@ -28,6 +28,8 @@ def download_url(url, model_name, output_path):
if os.path.exists(output_path): if os.path.exists(output_path):
return return
os.makedirs(os.path.expanduser("~/.u2net"), exist_ok=True)
print( print(
f"Downloading model to {output_path}".format(output_path=output_path), f"Downloading model to {output_path}".format(output_path=output_path),
file=sys.stderr, file=sys.stderr,