File size: 663 Bytes
26257ef
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
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()