Update README.md

README.md

@@ -16,8 +16,8 @@ tags:
 
 <!-- Provide the basic links for the model. -->
 
-- **Repository:** https://github.com/ninglab/GeLLMO
-- **Paper:** 
+- **Repository:** [Github](https://github.com/ninglab/GeLLMO-C)
+- **Paper:** [Arxiv](https://arxiv.org/abs/2505.23987)
 
 ## Usage
 
@@ -37,4 +37,19 @@ safety filters (e.g., toxicity or compliance checks).
 It is the responsibility of users to validate generated molecules for safety and ethical considerations.
 
 We urge users to adopt best practices, including toxicity prediction pipelines, 
-ethical oversight, and responsible AI usage policies, to prevent harmful applications of this model.
+ethical oversight, and responsible AI usage policies, to prevent harmful applications of this model.
+
+## Citation
+
+If you use the model in any form, please cite our paper as:
+```
+@article{dey2025llm,
+      title={Large Language Models for Controllable Multi-property Multi-objective Molecule Optimization}, 
+      author={Vishal Dey and Xiao Hu and Xia Ning},
+      year={2025},
+      eprint={2505.23987},
+      archivePrefix={arXiv},
+      primaryClass={cs.LG},
+      url={https://arxiv.org/abs/2505.23987}, 
+}
+```