Update README.md

README.md

@@ -71,11 +71,14 @@ ate_hat = causalpfn_ate.estimate_ate()
 If you use this model in your research, please cite:
 
 ```
-@misc{causalpfn2025,
-      title={CausalPFN: Amortized Causal Effect Estimation via In-Context Learning},
+@misc{balazadeh2025causalpfn,
+      title={CausalPFN: Amortized Causal Effect Estimation via In-Context Learning}, 
       author={Vahid Balazadeh and Hamidreza Kamkari and Valentin Thomas and Benson Li and Junwei Ma and Jesse C. Cresswell and Rahul G. Krishnan},
       year={2025},
+      eprint={2506.07918},
+      archivePrefix={arXiv},
       primaryClass={cs.LG},
+      url={https://arxiv.org/abs/2506.07918}, 
 }
 ```