Update README.md

README.md

@@ -12,6 +12,15 @@ tokenizer = AutoTokenizer.from_pretrained("Fujitsu/AugCode")
 model = AutoModelForSequenceClassification.from_pretrained("Fujitsu/AugCode")
 
 ```
-Then you may use it to infer the similarity between a given docstring and code.
-
+Then you may use `model` to infer the similarity between a given docstring and code.
 
+### Citation
+```bibtex@misc{bahrami2021augcode,
+    title={AugmentedCode: Examining the Effects of Natural Language Resources in Code Retrieval Models},
+    author={Mehdi Bahramix, N. C. Shrikanthy, Yuji Mizobuchiz, Lei Liux, Masahiro Fukuyoriz, Wei-Peng Chenx, Kazuki Munakataz},
+    year={2021},
+    eprint={TBA},
+    archivePrefix={TBA},
+    primaryClass={cs.CL}
+}
+```