Update entry_point.py

deep-diver · Jul 6, 2023 · 8261ba7 · 8261ba7
1 parent f6d5087
commit 8261ba7
Showing 1 changed file with 9 additions and 2 deletions.
diff --git a/entry_point.py b/entry_point.py
@@ -8,9 +8,16 @@
     parser = argparse.ArgumentParser()
 
     app_mode = os.getenv("LLMCHAT_APP_MODE")
+    local_files_only = os.getenv("LLMCHAT_LOCAL_FILES_ONLY")
+
     if app_mode is None or \
         app_mode not in ["GRADIO", "DISCORD"]:
         app_mode = "GRADIO"
+
+    if local_files_only is None:
+        local_files_only = False
+    else:
+        local_files_only = bool(local_files_only)
 
     if app_mode == "GRADIO":
         parser.add_argument('--root-path', default="")
@@ -29,6 +36,6 @@
         parser.add_argument('--mode-8bit', default=False, action=argparse.BooleanOptionalAction)
         parser.add_argument('--mode-4bit', default=False, action=argparse.BooleanOptionalAction)
         parser.add_argument('--mode-full-gpu', default=True, action=argparse.BooleanOptionalAction)
-        parser.add_argument('--local-files-only', default=False, action=argparse.BooleanOptionalAction)
+        parser.add_argument('--local-files-only', default=local_files_only, action=argparse.BooleanOptionalAction)
         args = parser.parse_args()
-        discord_main(args)
+        discord_main(args)