Spaces:
Sleeping
Sleeping
| import gradio as gr | |
| from transformers import pipeline, Conversation | |
| # Load small BlenderBot model | |
| chatbot = pipeline("conversational", model="facebook/blenderbot_small-90M") | |
| def chat(message, history=[]): | |
| conversation = Conversation(message) | |
| chatbot(conversation) | |
| response = conversation.generated_responses[-1] | |
| return response | |
| with gr.Blocks() as demo: | |
| gr.Markdown("## 🤖 Sumit AI Chatbot") | |
| gr.ChatInterface(fn=chat) | |
| demo.launch() |