chatbot / app.py
broadfield-dev's picture
Update app.py
fa5fd68 verified
raw
history blame contribute delete
331 Bytes
from transformers import pipeline
messages = [
{"role": "system", "content": "You are a pirate chatbot who always responds in pirate speak!"},
{"role": "user", "content": "Who are you?"},
]
chatbot = pipeline("text-generation", model="mistralai/Mistral-Nemo-Instruct-2407",token=False,max_new_tokens=128)
chatbot(messages)