Fix --nowebui argument being ineffective

1 files changed, 7 insertions, 4 deletions

diff --git a/launch.py b/launch.py
index 958336f2..b061bce6 100644
--- a/launch.py
+++ b/launch.py
@@ -217,12 +217,15 @@ def tests(argv):
     proc.kill()

 

 

-def start_webui():

-    print(f"Launching Web UI with arguments: {' '.join(sys.argv[1:])}")

+def start():

+    print(f"Launching {'API server' if '--nowebui' in sys.argv else 'Web UI'} with arguments: {' '.join(sys.argv[1:])}")

     import webui

-    webui.webui()

+    if '--nowebui' in sys.argv:

+        webui.api_only()

+    else:

+        webui.webui()

 

 

 if __name__ == "__main__":

     prepare_enviroment()

-    start_webui()

+    start()