From 332c16b58e0194f0960f9d6cf38746e7be8546b7 Mon Sep 17 00:00:00 2001 From: Daniel Gatis Date: Fri, 13 Nov 2020 13:47:24 -0300 Subject: [PATCH] fix typo --- setup.py | 2 +- src/rembg/cmd/cli.py | 6 ++++-- 2 files changed, 5 insertions(+), 3 deletions(-) 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: