Update README.md

README.md

@@ -13,4 +13,19 @@ This generative model is trained on a large data set (38k) of short single-parti
 
 ## Acknowledgements
 
-This project is funded by the National Science Foundation Division of Chemical, Bioengineering, Environmental, and Transport Systems under award 2338466, and the American Chemical Society Petroleum Research Fund under award 67239-DNI5.
+This project is funded by the National Science Foundation Division of Chemical, Bioengineering, Environmental, and Transport Systems under award 2338466, and the American Chemical Society Petroleum Research Fund under award 67239-DNI5.
+
+## Citation
+If you are using this code, please reference our paper:
+```
+  @article{shabeeb2025learning,
+  title={Learning the diffusion of nanoparticles in liquid phase TEM via physics-informed generative AI},
+  author={Shabeeb, Zain and Goyal, Naisargi and Attah Nantogmah, Pagnaa and Jamali, Vida},
+  journal={Nature Communications},
+  volume={16},
+  number={1},
+  pages={6298},
+  year={2025},
+  publisher={Nature Publishing Group UK London}
+}
+```