add web demo

Former-commit-id: 25ea647e5ac36b497b8e176b123fdee39be3fd30
This commit is contained in:
hiyouga
2023-07-18 17:21:16 +08:00
parent 5ed2a97056
commit 8e4ae0aaac
4 changed files with 59 additions and 4 deletions

View File

@@ -1,4 +1,4 @@
from typing import Dict, Tuple
from typing import Dict, Optional, Tuple
import gradio as gr
from gradio.blocks import Block
@@ -8,9 +8,10 @@ from llmtuner.webui.chat import WebChatModel
def create_chat_box(
chat_model: WebChatModel
chat_model: WebChatModel,
visible: Optional[bool] = False
) -> Tuple[Block, Component, Component, Dict[str, Component]]:
with gr.Box(visible=False) as chat_box:
with gr.Box(visible=visible) as chat_box:
chatbot = gr.Chatbot()
with gr.Row():