diff --git a/setup.py b/setup.py index 1452583..97a7051 100644 --- a/setup.py +++ b/setup.py @@ -11,7 +11,7 @@ with open("requirements.txt") as f: setup( name="rembg", - version="1.0.15", + version="1.0.16", description="Remove image background", long_description=long_description, long_description_content_type="text/markdown", diff --git a/src/rembg/cmd/cli.py b/src/rembg/cmd/cli.py index 96525cc..f0b89c4 100644 --- a/src/rembg/cmd/cli.py +++ b/src/rembg/cmd/cli.py @@ -3,6 +3,7 @@ import glob import os import filetype from distutils.util import strtobool +from tqdm import tqdm from ..bg import remove @@ -89,11 +90,12 @@ def main(): else: full_paths += glob.glob(path + "/*") - for fi in files: + for fi in tqdm(files): fi_type = filetype.guess(fi) + if fi_type is None: continue - elif if_type.mime.find('image') < 0: + elif fi_type.mime.find('image') < 0: continue with open(fi, "rb") as input: