Browse Source

fix(run-cli.py): support .webp files with --folder option

pull/105/head
Tom Brewe 12 months ago committed by GitHub
parent
commit
60294ebbe7
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
  1. 2
      run_cli.py

2
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')

Loading…
Cancel
Save