KleinZxA/testAI
0
1from transformers import pipeline, Conversation2import gradio as gr3bot = pipeline(model="openaccess-ai-collective/wizard-mega-13b")4 5message_list = []6response_list = []7 8def start(message, history):9 conversation = Conversation(text=message, past_user_inputs=message_list, generated_responses=response_list)10 conversation = bot(conversation)11 12 return conversation.generated_responses[-1]13 14demo_chatbot = gr.ChatInterface(start, title="XzierrAI", description="Enter text to start chatting.")15 16demo_chatbot.launch()