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