Update README.md

README.md

@@ -22,4 +22,16 @@ from transformers import AutoTokenizer, AutoModel
 
 tokenizer = AutoTokenizer.from_pretrained("Fujitsu/pytorrent")
 model = AutoModel.from_pretrained("Fujitsu/pytorrent")
+```
+## Citation
+```
+@misc{bahrami2021pytorrent,
+      title={PyTorrent: A Python Library Corpus for Large-scale Language Models}, 
+      author={Mehdi Bahrami and N. C. Shrikanth and Shade Ruangwan and Lei Liu and Yuji Mizobuchi and Masahiro Fukuyori and Wei-Peng Chen and Kazuki Munakata and Tim Menzies},
+      year={2021},
+      eprint={2110.01710},
+      archivePrefix={arXiv},
+      primaryClass={cs.SE},
+      howpublished={https://arxiv.org/abs/2110.01710},
+}
 ```