Update README.md

README.md

@@ -26,7 +26,7 @@ model = BartForConditionalGeneration.from_pretrained("HIT-TMG/dialogue-bart-larg
 
 # an example from CPC dev data
 history = ["可以 认识 一下 吗 ？", "当然 可以 啦 ， 你好 。", "嘿嘿 你好 ， 请问 你 最近 在 忙 什么 呢 ？", "我 最近 养 了 一只 狗狗 ， 我 在 训练 它 呢 。"]
-history_str = "历史：" + tokenizer.sep_token.join(history)
+history_str = "对话历史：" + tokenizer.sep_token.join(history)
 input_ids = tokenizer(history_str, return_tensors='pt').input_ids
 output_ids = model.generate(input_ids)[0]
 print(tokenizer.decode(output_ids))