File size: 588 Bytes
f22753e
7c585ce
 
1d8c407
 
7c585ce
 
f22753e
38e406c
 
 
 
 
f22753e
38e406c
 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
import gradio as gr
from transformers import pipeline, Conversation

chatbot = pipeline(task = 'conversational' , model = "facebook/blenderbot-400M-distill")

message_list = []
response_list = []

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()