Update README.md

README.md

@@ -16,16 +16,16 @@ from transformers.models.blenderbot import BlenderbotTokenizer, BlenderbotForCon
 def _norm(x):
 return ' '.join(x.strip().split())
 
-tokenizer = BlenderbotTokenizer.from_pretrained('thu-coai/blenderbot-400M-esconv')
-model = BlenderbotForConditionalGeneration.from_pretrained('thu-coai/blenderbot-400M-esconv')
+tokenizer = BlenderbotTokenizer.from_pretrained('chujiezheng/blenderbot-1B-augesc')
+model = BlenderbotForConditionalGeneration.from_pretrained('chujiezheng/blenderbot-1B-augesc')
 model.eval()
 
 utterances = [
-"I am having a lot of anxiety about quitting my current job. It is too stressful but pays well",
-"What makes your job stressful for you?",
-"I have to deal with many people in hard financial situations and it is upsetting",
-"Do you help your clients to make it to a better financial situation?",
-"I do, but often they are not going to get back to what they want. Many people are going to lose their home when safeguards are lifted",
+  "I am having a lot of anxiety about quitting my current job. It is too stressful but pays well",
+  "What makes your job stressful for you?",
+  "I have to deal with many people in hard financial situations and it is upsetting",
+  "Do you help your clients to make it to a better financial situation?",
+  "I do, but often they are not going to get back to what they want. Many people are going to lose their home when safeguards are lifted",
 ]
 input_sequence = ' '.join([' ' + e for e in utterances]) + tokenizer.eos_token # add space prefix and separate utterances with two spaces
 input_ids = tokenizer.convert_tokens_to_ids(tokenizer.tokenize(input_sequence))[-128:]
@@ -38,3 +38,31 @@ print(generation)
 
 utterances.append(generation[0]) # for future loop
 ```
+
+
+Please kindly cite the [original paper](https://aclanthology.org/2021.acl-long.269/) if you use this model:
+
+```bib
+@inproceedings{liu-etal-2021-towards,
+  title={Towards Emotional Support Dialog Systems},
+  author={Liu, Siyang  and 
+    Zheng, Chujie  and 
+    Demasi, Orianna  and 
+    Sabour, Sahand  and 
+    Li, Yu  and 
+    Yu, Zhou  and 
+    Jiang, Yong  and 
+    Huang, Minlie},
+  booktitle={Proceedings of the 59th annual meeting of the Association for Computational Linguistics},
+  year={2021}
+}
+@article{zheng2022augesc,
+  title={AugESC: Large-scale Data Augmentation for Emotional Support Conversation with Pre-trained Language Models},
+  author={Zheng, Chujie and
+    Sabour, Sahand and
+    Wen, Jiaxin and
+    Huang, Minlie},
+  journal={arXiv preprint arXiv:2202.13047},
+  year={2022}
+}
+```