|
import gradio as gr |
|
from gradio.themes.utils import colors |
|
from t5 import T5 |
|
|
|
def prepare_theme(): |
|
theme = gr.themes.Default(primary_hue=colors.gray, |
|
secondary_hue=colors.emerald, |
|
neutral_hue=colors.emerald).set( |
|
body_background_fill="*primary_800", |
|
body_background_fill_dark="*primary_800", |
|
|
|
block_background_fill="*primary_700", |
|
block_background_fill_dark="*primary_700", |
|
|
|
border_color_primary="*secondary_300", |
|
border_color_primary_dark="*secondary_300", |
|
block_border_width="3px", |
|
input_border_width="2px", |
|
|
|
input_background_fill="*primary_700", |
|
input_background_fill_dark="*primary_700", |
|
|
|
background_fill_secondary="*primary_700", |
|
background_fill_secondary_dark="*primary_700", |
|
|
|
body_text_color="white", |
|
body_text_color_dark="white", |
|
|
|
block_label_text_color="*secondary_300", |
|
block_label_text_color_dark="*secondary_300", |
|
block_label_background_fill="*primary_800", |
|
block_label_background_fill_dark="*primary_800", |
|
|
|
color_accent_soft="*primary_600", |
|
color_accent_soft_dark="*primary_600", |
|
) |
|
return theme |
|
|
|
def chat(message, chat_history, model): |
|
response = model.chat(message) |
|
chat_history.append((message, response)) |
|
return "", chat_history |
|
|
|
if __name__=='__main__': |
|
theme = prepare_theme() |
|
model = T5() |
|
|
|
with gr.Blocks(theme=theme) as demo: |
|
with gr.Row(): |
|
with gr.Column(scale=5): |
|
chatbot = gr.Chatbot(label="T5", bubble_full_width=False) |
|
with gr.Row(): |
|
txt = gr.Textbox(show_label=False, placeholder='Send a message...', container=False) |
|
|
|
txt.submit(chat, [txt, chatbot], [txt, chatbot]) |
|
|
|
demo.launch(debug=True, share=True) |