From 60294ebbe72eb2d24563c762e28ea445bba6bab6 Mon Sep 17 00:00:00 2001 From: Tom Brewe Date: Fri, 24 Nov 2023 19:43:09 +0100 Subject: [PATCH] fix(run-cli.py): support .webp files with --folder option --- run_cli.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/run_cli.py b/run_cli.py index b1a5ef7..9e708fa 100755 --- a/run_cli.py +++ b/run_cli.py @@ -73,7 +73,7 @@ def main(): print(f'The folder {args.folder} does not exist!') exit(1) - files = [f for f in os.listdir(args.folder) if f.endswith('.jpg') or f.endswith('.png')] + files = [f for f in os.listdir(args.folder) if f.endswith('.jpg') or f.endswith('.png') or f.endswith('webp')] prompts = [] for file in files: image = Image.open(os.path.join(args.folder, file)).convert('RGB')