Update README.md

README.md

@@ -6,7 +6,7 @@ language:
 - en
 pipeline_tag: sentence-similarity
 ---
-## Model to score relative persuasive language between pairs
+# Model to score relative persuasive language between pairs
 More info about training, evaluation, and use in paper is here: https://arxiv.org/abs/2406.17753
 
 Python:
@@ -29,4 +29,19 @@ def predict(textA, textB, model,tokenizer):
     score2=logits.detach().cpu().numpy()*(-1)
     score = (score1+score2)/2
     return score
+```
+
+## Citation
+
+If you find our dataset helpful, kindly refer to us in your work using the following citation:
+```
+@misc{pauli2024measuringbenchmarkinglargelanguage,
+      title={Measuring and Benchmarking Large Language Models' Capabilities to Generate Persuasive Language}, 
+      author={Amalie Brogaard Pauli and Isabelle Augenstein and Ira Assent},
+      year={2024},
+      eprint={2406.17753},
+      archivePrefix={arXiv},
+      primaryClass={cs.CL},
+      url={https://arxiv.org/abs/2406.17753}, 
+}
 ```