Update README.md

README.md

@@ -33,7 +33,7 @@ GENIE is co-developed by the groups of Sheng Yu (https://www.stat.tsinghua.edu.c
 ```python
 from vllm import LLM, SamplingParams
 
-PROMPT_TEMPLATE = "Human:\n{query}\n\n Assistant:"
+PROMPT_TEMPLATE = "Human:\n{query}\n\n Assistant:\n"
 sampling_params = SamplingParams(temperature=temperature, max_tokens=max_new_token)
 EHR = ['xxxxx1','xxxxx2']
 texts = [PROMPT_TEMPLATE.format(query=k) for k in EHR]