Update README.md

README.md

@@ -90,7 +90,7 @@ Here is a summary of the model hyper-parameters:
     tokenizer = AutoTokenizer.from_pretrained(model_id)
     model = AutoModelForCausalLM.from_pretrained(model_id)
     
-    text = '<|im_start|>user\nTranslate the following English source text to Portuguese:\nEnglish: I am a language model for european languages. \nPortuguese: <|im_end|>\n<|im_start|>assistant\n'
+    text = '<|im_start|>system\n<|im_end|>\n<|im_start|>user\nTranslate the following English source text to Portuguese:\nEnglish: I am a language model for european languages. \nPortuguese: <|im_end|>\n<|im_start|>assistant\n'
     
     inputs = tokenizer(text, return_tensors="pt")
     outputs = model.generate(**inputs, max_new_tokens=20)