fix args.input isdir

2023-01-08 21:26:57 +08:00 · 2023-01-08 21:26:57 +08:00 · bfb69347c2
commit bfb69347c2
parent aa5ace4099
1 changed files with 1 additions and 1 deletions
--- 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