Update README.md

README.md

@@ -1,7 +1,5 @@
 ---
 license: apache-2.0
-language:
-- en
 pipeline_tag: image-segmentation
 tags:
 - medical
@@ -15,6 +13,7 @@ tags:
 - 7T
 - Conditional VAE
 - distribution distance
+library_name: pytorch
 ---
 # PULASki_ProbUNet2D_Sinkhorn_VSeg
 
@@ -24,6 +23,9 @@ We proposed the PULASki as a computationally efficient generative tool for biome
 
 ## Model Details
 
+It was introduced in [PULASki: Learning inter-rater variability using statistical distances to improve
+probabilistic segmentation](https://arxiv.org/abs/2312.15686) by Soumick Chatterjee, Franziska Gaidzik, Alessandro Sciarra, Hendrik Mattern, Gábor Janiga, Oliver Speck, Andreas Nürnberger, Sahani Pathiraja.
+
 ### Model Description
 
 - **Developed by:** Dr Soumick Chatterjee