Update README.md

README.md

@@ -58,15 +58,15 @@ Em apoio aos países de língua portuguesa.
 import torch
 from transformers import pipeline
 
-samba = pipeline('text-generation', model='lrds-code/samba-1.1B', torch_dtype=torch.bfloat16, device_map='auto')
+boana = pipeline('text-generation', model='lrds-code/boana-7b-instruct', torch_dtype=torch.bfloat16, device_map='auto')
 
 messages = [{'role':'system',
              'content':''},
             {'role':'user',
              'content':'Quantos planetas existem no sistema solar?'}]
 
-prompt = samba.tokenizer.apply_chat_template(messages, tokenize=False, add_generation_prompt=True)
-outputs = samba(prompt, max_new_tokens=256, do_sample=False, temperature=0.1, top_k=50, top_p=0.95, repetition_penalty=1.1, do_sample=False)
+prompt = boana.tokenizer.apply_chat_template(messages, tokenize=False, add_generation_prompt=True)
+outputs = boana(prompt, max_new_tokens=256, do_sample=False, temperature=0, top_k=50, top_p=0.95)
 print(outputs[0]['generated_text'])
 
 # <s>[INST] <<SYS>>