Spaces:
Sleeping
Sleeping
File size: 703 Bytes
b69ac7d 861b37b 62cc9fa 861b37b b69ac7d 861b37b 03c843d 4f57f80 b69ac7d 861b37b 4f57f80 f969418 355a4ae b69ac7d 861b37b 03c843d b69ac7d |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 |
from transformers import pipeline, Conversation
import gradio as gr
chatbot = pipeline(model="facebook/blenderbot-400M-distill", task="conversational")
message_list = []
response_list = []
def Stark_chatbot(message, history):
global message_list, response_list
message_list.append(message)
conversation = Conversation(text=message, past_user_inputs=message_list[:-1], generated_responses=response_list)
conversation = chatbot(conversation)
reply = conversation.generated_responses[-1]
response_list.append(reply)
return reply
demo_chatbot = gr.ChatInterface(Stark_chatbot, title="Stark Chatbot", description="Enter text to start chatting.")
demo_chatbot.launch()
|