Update README.md

README.md

@@ -1,4 +1,4 @@
-# Discourse marker / connective prediction
+# Discourse marker / connective prediction pretrained model
 
 `roberta-base` pretrained on discourse marker prediction on the Discovery dataset with a validation accuracy of 30.93% (majority class is 0.57%)
 
@@ -6,4 +6,24 @@ https://github.com/sileod/discovery
 
 https://huggingface.co/datasets/discovery
 
-This model can be used as a pretrained model for discourse tasks
+This model can also be used as a pretrained model for NLU, pragmatics and discourse tasks
+
+## Citing & Authors
+
+```bibtex 
+@inproceedings{sileo-etal-2019-mining,
+    title = "Mining Discourse Markers for Unsupervised Sentence Representation Learning",
+    author = "Sileo, Damien  and
+      Van De Cruys, Tim  and
+      Pradel, Camille  and
+      Muller, Philippe",
+    booktitle = "Proceedings of the 2019 Conference of the North {A}merican Chapter of the Association for Computational Linguistics: Human Language Technologies, Volume 1 (Long and Short Papers)",
+    month = jun,
+    year = "2019",
+    address = "Minneapolis, Minnesota",
+    publisher = "Association for Computational Linguistics",
+    url = "https://aclanthology.org/N19-1351",
+    doi = "10.18653/v1/N19-1351",
+    pages = "3477--3486",
+}
+```