|
|
import gradio as gr |
|
|
from transformers import pipeline |
|
|
|
|
|
|
|
|
model = pipeline("text-generation", model="gpt2") |
|
|
|
|
|
def respond(message, chat_history): |
|
|
|
|
|
response = model(message, max_length=50)[0]["generated_text"] |
|
|
chat_history.append({"role": "user", "content": message}) |
|
|
chat_history.append({"role": "assistant", "content": response}) |
|
|
return "", chat_history |
|
|
|
|
|
with gr.Blocks() as demo: |
|
|
chatbot = gr.Chatbot(type="messages") |
|
|
msg = gr.Textbox(label="Type your message") |
|
|
clear = gr.ClearButton([msg, chatbot]) |
|
|
msg.submit(respond, [msg, chatbot], [msg, chatbot]) |
|
|
|
|
|
demo.launch() |
|
|
|
