Spaces:
Runtime error
Runtime error
Update cli_batch_app.py
Browse files- cli_batch_app.py +1 -1
cli_batch_app.py
CHANGED
|
@@ -381,7 +381,7 @@ def main(
|
|
| 381 |
|
| 382 |
# Check if input_path is a directory or a single image
|
| 383 |
if osp.isdir(input_path):
|
| 384 |
-
image_paths = [osp.join(input_path, fname) for fname in os.listdir(input_path) if fname.lower().endswith(('.png', '.jpg', '.jpeg'))]
|
| 385 |
else:
|
| 386 |
image_paths = [input_path]
|
| 387 |
|
|
|
|
| 381 |
|
| 382 |
# Check if input_path is a directory or a single image
|
| 383 |
if osp.isdir(input_path):
|
| 384 |
+
image_paths = [osp.join(input_path, fname) for fname in os.listdir(input_path) if fname.lower().endswith(('.png', '.jpg', '.jpeg', ".webp"))]
|
| 385 |
else:
|
| 386 |
image_paths = [input_path]
|
| 387 |
|