File size: 604 Bytes
6a91d6f 4636668 ebe1aed 6a91d6f ebe1aed 6a91d6f 4636668 6a91d6f d1262ae 4636668 6a91d6f |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 |
from transformers import pipeline
from transformers import Conversation
import gradio as gr
message_list = []
response_list = []
chatbot = pipeline("conversational", model = "facebook/blenderbot-400M-distill")
def vanilla_chatbot(message, history):
conversation = Conversation(text = message, past_user_inputs = message_list, generated_responses = response_list)
conversation = chatbot(conversation)
return conversation.generated_responses[-1]
demo_chatbot = gr.ChatInterface(vanilla_chatbot, title = "Vanilla Chatbot", description = "Enter text to start chatting.")
demo_chatbot.launch() |