File size: 631 Bytes
2918e76
 
 
 
 
 
 
 
 
 
 
 
 
362c65c
 
 
 
 
2918e76
 
 
 
362c65c
 
 
 
 
2918e76
 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
from transformers import pipeline
from transformers import Conversation

import gradio as gr

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

import gradio as gr

message_list = []
response_list = []

def mini_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(
    mini_chatbot,
    title="My Chatbot",
    description="Enter text to start chatting."
)

demo_chatbot.launch()