Update README.md

README.md

@@ -78,4 +78,17 @@ import torch
 '[C][N][O][/C][=C][/C][=C][C][=Branch1][C][=O][C][=C]',
 '[C][N][O][C][=Branch1][C][=O][N][Branch1][C][C][C][=Branch1]',
 '[C][N][O][Ring1][Branch1][C][C][C][C][C][C][C][C]']
-```
+```
+
+## Citation
+If you use our repository, please cite:
+```bibtex
+@article{molgen-7b,
+  author = {Yin Fang, Xiaozhuan Liang, Ningyu Zhang},
+  title = {MolGen-7b},
+  year = {2023},
+  publisher = {GitHub},
+  journal = {GitHub repository},
+  howpublished = {\url{https://github.com/}},
+}
+```