fix auto save to output_dir

2024-02-09 15:17:28 +08:00
parent 314b2ae8c0
commit 131fa8300b
4 changed files with 11 additions and 9 deletions
--- a/iopaint/cli.py
+++ b/iopaint/cli.py
@@ -144,7 +144,10 @@ def start(
    device = check_device(device)
    if input and not input.exists():
        logger.error(f"invalid --input: {input} not exists")
-        exit()
+        exit(-1)
+    if input and input.is_dir() and not output_dir:
+        logger.error(f"invalid --output-dir: must be set when --input is a directory")
+        exit(-1)
    if output_dir:
        output_dir = output_dir.expanduser().absolute()
        logger.info(f"Image will be saved to {output_dir}")