Update README.md

README.md

@@ -110,14 +110,17 @@ generation_config = GenerationConfig(
 )
 
 
+
 def format_system_prompt(prompt, eos_token="</s>"):
-    return "{}{}{}".format(
+    return "{}{}{}{}".format(
         "<|prompter|>",
         prompt,
         eos_token,
+        "<|assistant|>"
     )
 
 
+
 def generate(prompt, generation_config=generation_config, max_new_tokens=2048, device=device):
     prompt = format_system_prompt(prompt)  # OpenAssistant Prompt Format expected
     input_ids = tokenizer(prompt, return_tensors="pt").input_ids.to(device)