Spaces:
Runtime error
Runtime error
| from transformers import pipeline, Conversation | |
| import gradio as gr | |
| Chatbot = pipeline(task = "conversational", model ="facebook/blenderbot-400M-distill") | |
| #text, get it from user from gradio | |
| message_list = [] | |
| response_list = [] | |
| def mini_chatbot(message, history): | |
| conversation = Conversation(text = message, | |
| past_user_inputs = message_list, | |
| generated_responses = response_list) | |
| response = Chatbot(conversation) | |
| return response.generated_responses[-1] | |
| demo_chatbot = gr.ChatInterface(mini_chatbot, title="AI Chatbot" , description="Enter text to CHAT with AI") | |
| demo_chatbot.launch() |