Update README.md

README.md

@@ -47,8 +47,8 @@ from transformers import AutoTokenizer, AutoModelForCausalLM
 tokenizer = AutoTokenizer.from_pretrained("Fsoft-AIC/XMAiNframe-instruct-10.5b")
 model = AutoModelForCausalLM.from_pretrained("Fsoft-AIC/XMAiNframe-instruct-10.5b")
 messages=[
-    {'role':'system','content':"You are a helpful assistant"},
-    {'role': 'user', 'content': 'What is the future of Mainframe?'}
+    {'from':'system', 'value': "You are a helpful assistant"},
+    {'from': 'human', 'value': 'What is the future of Mainframe?'}
 ]
 inputs = tokenizer.apply_chat_template(messages, add_generation_prompt=True, return_tensors="pt").to(model.device)