Update README.md

README.md

@@ -27,7 +27,12 @@ model = AutoModelForCausalLM.from_pretrained(model_name, torch_dtype=torch.float
 tokenizer = AutoTokenizer.from_pretrained(model_name)
 
 messages = [
-    {'role': 'system', 'content': "You are a helpful and harmless AI assistant, developed by Samsung"}
+    {'role': 'system', 'content': """You are a helpful and harmless AI assistant.
+A conversation between User and Assistant. The user asks a question, and the Assistant solves it.  
+The assistant first thinks about the reasoning process in the mind and then provides the user with the answer.  
+The reasoning process and answer are enclosed within `<think> </think>` and `<answer> </answer>` tags, respectively, i.e., `<think> reasoning process here </think>` `<answer> answer here </answer>`.  
+User: **prompt**.  
+Assistant:"""}
 ]
 
 def stream_output(output_text):