kyy/llm_trainer
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

update web demo

Browse Source
This commit is contained in:
hiyouga
2023-08-03 13:28:28 +08:00
parent e23a3a366c
commit 53d6dc396d
8 changed files with 31 additions and 48 deletions
Download Patch File Download Diff File Expand all files Collapse all files

21
src/llmtuner/webui/interface.py
View File

@@ -6,8 +6,10 @@ from llmtuner.webui.components import (
create_sft_tab,
create_eval_tab,
create_infer_tab,
create_export_tab
create_export_tab,
create_chat_box
)
from llmtuner.webui.chat import WebChatModel
from llmtuner.webui.css import CSS
from llmtuner.webui.manager import Manager
from llmtuner.webui.runner import Runner
@@ -53,6 +55,23 @@ def create_ui() -> gr.Blocks:
return demo
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="en")
_, _, _, chat_elems = create_chat_box(chat_model, visible=True)
manager = Manager([{"lang": lang}, chat_elems])
demo.load(manager.gen_label, [lang], [lang] + list(chat_elems.values()))
lang.change(manager.gen_label, [lang], [lang] + list(chat_elems.values()))
return demo
if __name__ == "__main__":
demo = create_ui()
demo.queue()