Update README.md

README.md

@@ -22,11 +22,20 @@ model = "recogna-nlp/phibode-3-mini-4k-ultraalpaca"
 tokenizer = AutoTokenizer.from_pretrained(model)
 
 # Example prompt
-prompt = "Your example prompt here"
+messages = [
+    {"role": "system", "content": "Você é assistente de IA chamado PhiBode. O PhiBode é um modelo de língua conversacional projetado para ser prestativo, honesto e inofensivo."},
+    {"role": "user", "content": "<Insira seu prompt aqui>"},
+]
 
 # Generate a response
-model = AutoModelForCausalLM.from_pretrained(model)
+model = AutoModelForCausalLM.from_pretrained(model, trust_remote_code=True)
 pipeline = pipeline("text-generation", model=model, tokenizer=tokenizer)
-outputs = pipeline(prompt, max_length=50, num_return_sequences=1)
+generation_args = {
+    "max_new_tokens": 500,
+    "return_full_text": False,
+    "temperature": 0.0,
+    "do_sample": False,
+}
+outputs = pipeline(messages, **generation_args)
 print(outputs[0]["generated_text"])
 ```