File size: 331 Bytes
70dd54a
8014031
 
70dd54a
 
8014031
fa5fd68
70dd54a
1
2
3
4
5
6
7
8
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)