From f4bf49e891728bf28e5c1aed48210b7796420eef Mon Sep 17 00:00:00 2001 From: hiyouga <467089858@qq.com> Date: Fri, 17 May 2024 00:08:56 +0800 Subject: [PATCH] enable inbrowser in webui Former-commit-id: 694a05fd044bbbad107ca8fed5494460c78e1981 --- src/llamafactory/webui/interface.py | 4 ++-- src/webui.py | 2 +- 2 files changed, 3 insertions(+), 3 deletions(-) diff --git a/src/llamafactory/webui/interface.py b/src/llamafactory/webui/interface.py index c5a30113..bae3ba76 100644 --- a/src/llamafactory/webui/interface.py +++ b/src/llamafactory/webui/interface.py @@ -73,10 +73,10 @@ def create_web_demo() -> gr.Blocks: def run_web_ui() -> None: gradio_share = os.environ.get("GRADIO_SHARE", "0").lower() in ["true", "1"] server_name = os.environ.get("GRADIO_SERVER_NAME", "0.0.0.0") - create_ui().queue().launch(share=gradio_share, server_name=server_name) + create_ui().queue().launch(share=gradio_share, server_name=server_name, inbrowser=True) def run_web_demo() -> None: gradio_share = os.environ.get("GRADIO_SHARE", "0").lower() in ["true", "1"] server_name = os.environ.get("GRADIO_SERVER_NAME", "0.0.0.0") - create_web_demo().queue().launch(share=gradio_share, server_name=server_name) + create_web_demo().queue().launch(share=gradio_share, server_name=server_name, inbrowser=True) diff --git a/src/webui.py b/src/webui.py index 8da69497..bbefb54e 100644 --- a/src/webui.py +++ b/src/webui.py @@ -6,7 +6,7 @@ from llamafactory.webui.interface import create_ui def main(): gradio_share = os.environ.get("GRADIO_SHARE", "0").lower() in ["true", "1"] server_name = os.environ.get("GRADIO_SERVER_NAME", "0.0.0.0") - create_ui().queue().launch(share=gradio_share, server_name=server_name) + create_ui().queue().launch(share=gradio_share, server_name=server_name, inbrowser=True) if __name__ == "__main__":