Update README.md

README.md

@@ -1,3 +1,22 @@
----
-license: mit
----
+---
+license: mit
+---
+
+# Models for FlashAffinity
+
+This repository hosts the pre-trained model checkpoints for the paper **"FlashAffinity: Bridging the Accuracy-Speed Gap in Protein-Ligand Binding Affinity Prediction"**.
+
+The main code repository can be found at [FlashAffinity](https://github.com/clorf6/FlashAffinity).
+
+## Citation
+
+If you use FlashAffinity in your research, please cite our paper:
+
+```bibtex
+@article {Jiang2025.12.22.695983,
+	author = {Jiang, Songlin and Chen, Yifan and Cao, Ze and Jin, Wengong},
+	title = {FlashAffinity: Bridging the Accuracy-Speed Gap in Protein-Ligand Binding Affinity Prediction},
+	year = {2025},
+	journal = {bioRxiv}
+}
+```