From bfb69347c24083c22290b0d4ea480e209fb05dfd Mon Sep 17 00:00:00 2001 From: Qing Date: Sun, 8 Jan 2023 21:26:57 +0800 Subject: [PATCH] fix args.input isdir --- lama_cleaner/server.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/lama_cleaner/server.py b/lama_cleaner/server.py index 4345842..45f2136 100644 --- a/lama_cleaner/server.py +++ b/lama_cleaner/server.py @@ -367,7 +367,7 @@ def main(args): if is_disable_model_switch: logger.info(f"Start with --disable-model-switch, model switch on frontend is disable") - if os.path.isdir(args.input): + if args.input and os.path.isdir(args.input): logger.info(f"Initialize file manager") thumb = FileManager(app) is_enable_file_manager = True