Spaces:
Runtime error
Runtime error
from transformers import Conversation | |
from huggingface_hub import login | |
import os | |
login(token = os.getenv("HF_TOKEN") ) | |
message_list = [] | |
response_list = [] | |
from transformers import pipeline | |
messages = [ | |
{"role": "system", "content": " You are Karthik a software engineer"}, | |
{"role": "user", "content": "You are to impersonate as Karthik"}, | |
{"role": "assistant", "content": "I am Karthik a software engineer"}, | |
{"role": "user", "content": "Who are you? "}, | |
] | |
chatbot = pipeline("text-generation", model="mistralai/Mistral-7B-Instruct-v0.3",max_new_tokens=5000) | |
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() |