Update README.md

README.md

@@ -50,4 +50,18 @@ print(tokenizer.decode(output))
 '''
 ```
 
-For more details of our work, please refer to our main [project repo](https://github.com/yxuansu/SimCTG).
+For more details of our work, please refer to our main [project repo](https://github.com/yxuansu/SimCTG).
+
+## 5. Citation:
+If you find our paper and resources useful, please kindly leave a star and cite our paper. Thanks!
+
+```bibtex
+@article{su2022contrastive,
+  title={A Contrastive Framework for Neural Text Generation},
+  author={Su, Yixuan and Lan, Tian and Wang, Yan and Yogatama, Dani and Kong, Lingpeng and Collier, Nigel},
+  journal={arXiv preprint arXiv:2202.06417},
+  year={2022}
+}
+```
+
+