Update README.md

README.md

@@ -45,11 +45,12 @@ streamer = TextStreamer(tokenizer,
 
 # Convert prompt to tokens
 prompt_template = """\
-<|im_start|>system
-{system_message}<|im_end|>
-<|im_start|>user
-{prompt}<|im_end|>
-<|im_start|>assistant"""
+### System:
+You are an AI assistant. User will give you a task. Your goal is to complete the task as faithfully as you can. While performing the task think step-by-step and justify your steps.
+### User: 
+{prompt} 
+### Assistant:
+"""
 
 prompt = "You're standing on the surface of the Earth. "\
         "You walk one mile south, one mile west and one mile north. "\