diff --git a/run.py b/run.py index 1cc9026..d90da33 100644 --- a/run.py +++ b/run.py @@ -5,9 +5,13 @@ from tagger.interrogator.interrogator import AbsInterrogator from PIL import Image from pathlib import Path import argparse +from PIL import ImageFile from tagger.interrogators import interrogators +# Allow images with broken headers to load +ImageFile.LOAD_TRUNCATED_IMAGES = True + parser = argparse.ArgumentParser() group = parser.add_mutually_exclusive_group(required=True)