File size: 436 Bytes
fd58f73
91f003e
 
224f52e
91f003e
fd58f73
 
 
224f52e
fd58f73
50444a1
fd58f73
224f52e
 
 
50444a1
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
from transformers import pipeline, Conversation
import gradio as gr

chatbot = pipeline(model="facebook/blenderbot-400M-distill")

message_list = []
response_list = []

def vanilla_chatbot(message, history):
    conversation = chatbot(message)
    
    return conversation[0]['generated_text']

demo_chatbot = gr.ChatInterface(vanilla_chatbot, title="Vanilla Chatbot", description="Enter text to start chatting.")

demo_chatbot.launch()