| import gradio as gr | |
| from transformers import pipeline | |
| chatbot = pipeline("text-generation", model="microsoft/DialoGPT-small") | |
| def respond(message): | |
| return chatbot(message, max_length=100)[0]["generated_text"] | |
| demo = gr.Interface(fn=respond, inputs="text", outputs="text") | |
| demo.launch() |