From 7ac9aebb135c88061000fb11c486273818a38dd8 Mon Sep 17 00:00:00 2001
From: Thibaud Coroller <14895385+tcoroller@users.noreply.github.com>
Date: Wed, 17 Apr 2024 10:03:49 -0400
Subject: [PATCH] Update README.md

---
 README.md | 15 ++++++++++++---
 1 file changed, 12 insertions(+), 3 deletions(-)

diff --git a/README.md b/README.md
index ae3eb7d..51d06bb 100644
--- a/README.md
+++ b/README.md
@@ -4,9 +4,9 @@
 
 ![CodeQC](https://github.com/Novartis/torchsurv/actions/workflows/codeqc.yml/badge.svg?branch=main)
 ![Docs](https://github.com/Novartis/torchsurv/actions/workflows/docs.yml/badge.svg?branch=main)
-[![arXiv](https://img.shields.io/badge/arXiv-2404.10761-f9f107.svg)](https://arxiv.org/abs/2404.10761)
-[![Documentation](https://img.shields.io/badge/GitBook-Docu-lightblue)](https://opensource.nibr.com/torchsurv/)
 [![PyPI version](https://badge.fury.io/py/torchsurv.svg)](https://badge.fury.io/py/torchsurv)
+[![arXiv](https://img.shields.io/badge/arXiv-2404.10761-f9f107.svg)](https://arxiv.org/abs/2404.10761)
+[![Documentation](https://img.shields.io/badge/GitBook-Sphinx-blue)](https://opensource.nibr.com/torchsurv/)
 [![Downloads](https://static.pepy.tech/badge/torchsurv)](https://pepy.tech/project/torchsurv)
 
 
@@ -224,4 +224,13 @@ If you have any questions, suggestions, or feedback, feel free to reach out the
 
 If you use this project in academic work or publications, we appreciate citing it using the following BibTeX entry:
 
-TEMPORARY PLACEHOLDER.
+```
+@misc{monod2024torchsurv,
+      title={TorchSurv: A Lightweight Package for Deep Survival Analysis}, 
+      author={Melodie Monod and Peter Krusche and Qian Cao and Berkman Sahiner and Nicholas Petrick and David Ohlssen and Thibaud Coroller},
+      year={2024},
+      eprint={2404.10761},
+      archivePrefix={arXiv},
+      primaryClass={cs.LG}
+}
+```