Update README.md

README.md

@@ -5,7 +5,9 @@ datasets:
 pipeline_tag: graph-ml
 ---
 
-Here is the pretrained checkpoint for "Learning Molecular Representation in a Cell," trained with the following configurations:
+The pretrained model for the paper ["Learning Molecular Representation in a Cell"](https://arxiv.org/abs/2406.12056v1). 
+
+The model is trained with the following configurations:
 
 - **emb_dim**: 300
 - **model**: gin-virtual