Update README.md

README.md

@@ -23,7 +23,7 @@ import transformers
 tokenizer = transformers.AutoTokenizer.from_pretrained("RekaAI/reka-flash-3")
 model = transformers.AutoModelForCausalLM.from_pretrained("RekaAI/reka-flash-3", torch_dtype='auto', device_map='auto')
 
-prompt = {"role": "user", "content": "Write a poem about large language model."}
+prompt = {"role": "human", "content": "Write a poem about large language model."}
 text = tokenizer.apply_chat_template([prompt], tokenize=False, add_generation_prompt=True)
 model_inputs = tokenizer([text], return_tensors="pt").to(model.device)
 outputs = model.generate(**model_inputs, max_new_tokens=512)