From ca08e5efd3ccf155c6a0baa0a08bb7879f6b9b9b Mon Sep 17 00:00:00 2001 From: hiyouga Date: Mon, 14 Aug 2023 11:37:01 +0800 Subject: [PATCH] fix webui cache --- src/llmtuner/webui/components/top.py | 2 +- src/llmtuner/webui/interface.py | 4 ++-- 2 files changed, 3 insertions(+), 3 deletions(-) diff --git a/src/llmtuner/webui/components/top.py b/src/llmtuner/webui/components/top.py index 8611e280..1f5c47a5 100644 --- a/src/llmtuner/webui/components/top.py +++ b/src/llmtuner/webui/components/top.py @@ -15,7 +15,7 @@ def create_top() -> Dict[str, "Component"]: available_models = list(SUPPORTED_MODELS.keys()) + ["Custom"] with gr.Row(): - lang = gr.Dropdown(choices=["en", "zh"], value="en", scale=1) + lang = gr.Dropdown(choices=["en", "zh"], scale=1) model_name = gr.Dropdown(choices=available_models, scale=3) model_path = gr.Textbox(scale=3) diff --git a/src/llmtuner/webui/interface.py b/src/llmtuner/webui/interface.py index 0ae59224..1a5c0c19 100644 --- a/src/llmtuner/webui/interface.py +++ b/src/llmtuner/webui/interface.py @@ -59,7 +59,7 @@ def create_web_demo() -> gr.Blocks: chat_model = WebChatModel(lazy_init=False) with gr.Blocks(title="Web Demo", css=CSS) as demo: - lang = gr.Dropdown(choices=["en", "zh"], value="") + lang = gr.Dropdown(choices=["en", "zh"]) _, _, _, chat_elems = create_chat_box(chat_model, visible=True) @@ -67,7 +67,7 @@ def create_web_demo() -> gr.Blocks: demo.load(manager.gen_label, [lang], [lang] + list(chat_elems.values())) - lang.change(manager.gen_label, [lang], [lang] + list(chat_elems.values()), queue=False) + lang.select(manager.gen_label, [lang], [lang] + list(chat_elems.values()), queue=False) return demo