Added citations

README.md

@@ -9,7 +9,7 @@ widget:
 - text: "Silva, ti si grda in neprijazna"
 ---
 
-Text classification model based on `EMBEDDIA/sloberta` and fine-tuned on the [FRANK dataset](https://www.clarin.si/repository/xmlui/handle/11356/1433) comprising of LGBT and migrant hatespeech. Only the slovenian subset of the data was used for fine-tuning and the dataset has been relabeled for binary classification (offensive or acceptable).
+Text classification model based on `EMBEDDIA/sloberta` and fine-tuned on the [FRENK dataset](https://www.clarin.si/repository/xmlui/handle/11356/1433) comprising of LGBT and migrant hatespeech. Only the slovenian subset of the data was used for fine-tuning and the dataset has been relabeled for binary classification (offensive or acceptable).
 
 ## Fine-tuning hyperparameters
 
@@ -69,4 +69,47 @@ predictions, logit_output = model.predict(["Silva, ti si grda in neprijazna", "N
 predictions
 ### Output:
 ### array([1, 0])
-```
+```
+
+## Citation
+
+If you use the model, please cite the following paper on which the original model is based:
+
+```
+@article{DBLP:journals/corr/abs-1907-11692,
+  author    = {Yinhan Liu and
+               Myle Ott and
+               Naman Goyal and
+               Jingfei Du and
+               Mandar Joshi and
+               Danqi Chen and
+               Omer Levy and
+               Mike Lewis and
+               Luke Zettlemoyer and
+               Veselin Stoyanov},
+  title     = {RoBERTa: {A} Robustly Optimized {BERT} Pretraining Approach},
+  journal   = {CoRR},
+  volume    = {abs/1907.11692},
+  year      = {2019},
+  url       = {http://arxiv.org/abs/1907.11692},
+  archivePrefix = {arXiv},
+  eprint    = {1907.11692},
+  timestamp = {Thu, 01 Aug 2019 08:59:33 +0200},
+  biburl    = {https://dblp.org/rec/journals/corr/abs-1907-11692.bib},
+  bibsource = {dblp computer science bibliography, https://dblp.org}
+}
+
+```
+
+and the dataset used for fine-tuning:
+
+```
+@misc{ljubešić2019frenk,
+      title={The FRENK Datasets of Socially Unacceptable Discourse in Slovene and English}, 
+      author={Nikola Ljubešić and Darja Fišer and Tomaž Erjavec},
+      year={2019},
+      eprint={1906.02045},
+      archivePrefix={arXiv},
+      primaryClass={cs.CL},
+      url={https://arxiv.org/abs/1906.02045}
+}