Update README.md

README.md

@@ -2,7 +2,7 @@
 
 This model is a scoring function for images generated from text. It takes as input a prompt and a generated image and outputs a score. 
 It can be used as a general scoring function, and for tasks such as human preference prediction, model evaluation, image ranking, and more. 
-See our paper (TODO) for more details.
+See our paper [Pick-a-Pic: An Open Dataset of User Preferences for Text-to-Image Generation](https://arxiv.org/abs/2305.01569) for more details.
 
 
 ## Model Details
@@ -16,7 +16,7 @@ This model was finetuned from CLIP-H using the [Pick-a-Pic dataset](https://hugg
 <!-- Provide the basic links for the model. -->
 
 - **Repository:** [See the PickScore repo](https://github.com/yuvalkirstain/PickScore)
-- **Paper [optional]:** TODO
+- **Paper:** [Pick-a-Pic: An Open Dataset of User Preferences for Text-to-Image Generation](https://arxiv.org/abs/2305.01569).
 - **Demo [optional]:** TODO
 
 ## How to Get Started with the Model
@@ -89,11 +89,18 @@ TODO - add paper.
 
 ## Citation [optional]
 
-TODO add paper
-
-**BibTeX:**
-
-[More Information Needed]
+If you find this work useful, please cite:
+
+```bibtex
+@misc{kirstain2023pickapic,
+      title={Pick-a-Pic: An Open Dataset of User Preferences for Text-to-Image Generation}, 
+      author={Yuval Kirstain and Adam Polyak and Uriel Singer and Shahbuland Matiana and Joe Penna and Omer Levy},
+      year={2023},
+      eprint={2305.01569},
+      archivePrefix={arXiv},
+      primaryClass={cs.CV}
+}
+```
 
 **APA:**