mirror of
https://git.mirrors.martin98.com/https://github.com/danielgatis/rembg
synced 2025-08-14 17:05:56 +08:00
fix issue #2
This commit is contained in:
parent
28100b3a52
commit
1478c5845c
2
setup.py
2
setup.py
@ -11,7 +11,7 @@ with open("requirements.txt") as f:
|
||||
|
||||
setup(
|
||||
name="rembg",
|
||||
version="1.0.5",
|
||||
version="1.0.6",
|
||||
description="Remove image background",
|
||||
long_description=long_description,
|
||||
long_description_content_type="text/markdown",
|
||||
|
@ -28,6 +28,8 @@ def download_url(url, model_name, output_path):
|
||||
if os.path.exists(output_path):
|
||||
return
|
||||
|
||||
os.makedirs(os.path.expanduser("~/.u2net"), exist_ok=True)
|
||||
|
||||
print(
|
||||
f"Downloading model to {output_path}".format(output_path=output_path),
|
||||
file=sys.stderr,
|
||||
|
Loading…
x
Reference in New Issue
Block a user