This commit is contained in:
Daniel Gatis 2020-11-13 13:47:24 -03:00
parent 4659d52273
commit 332c16b58e
2 changed files with 5 additions and 3 deletions

View File

@ -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",

View File

@ -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: