Update README.md

README.md

@@ -16,7 +16,7 @@ widget:
 
 # Emotion English DistilRoBERTa-base
 
-## Description ℹ
+# Description ℹ
 
 With this model, you can classify emotions in English text data. The model was trained on 6 diverse datasets (see Appendix below) and predicts Ekman's 6 basic emotions, plus a neutral class:
 
@@ -30,7 +30,7 @@ With this model, you can classify emotions in English text data. The model was t
 
 The model is a fine-tuned checkpoint of [DistilRoBERTa-base](https://huggingface.co/distilroberta-base). For a 'non-distilled' emotion model, please refer to the model card of the [RoBERTa-large](https://huggingface.co/j-hartmann/emotion-english-roberta-large) version.
 
-## Application 🚀
+# Application 🚀
 
 a) Run emotion model with 3 lines of code on single text example using Hugging Face's pipeline command on Google Colab:
 
@@ -40,13 +40,27 @@ b) Run emotion model on multiple examples and full datasets (e.g., .csv files) o
 
 [![Open In Colab](https://colab.research.google.com/assets/colab-badge.svg)](https://colab.research.google.com/github/j-hartmann/emotion-english-distilroberta-base/blob/main/emotion_prediction_example.ipynb)
 
-## Contact 💻
+# Contact 💻
 
 Please reach out to [j.p.hartmann@rug.nl](mailto:j.p.hartmann@rug.nl) if you have any questions or feedback.
 
 Thanks to Samuel Domdey and chrsiebert for their support in making this model available.
 
-## Appendix 📚
+# Reference
+
+Please cite the following reference if you use this model. A working paper will be available soon.
+
+```
+@misc{hartmann2022,
+  title={Emotion English DistilRoBERTa-base},
+  author={Hartmann, Jochen},
+  url={https://huggingface.co/j-hartmann/emotion-english-distilroberta-base/},
+  year={2022},
+  note={Online; accessed [INSERT_DATE]}
+}
+```
+
+# Appendix 📚
 
 Please find an overview of the datasets used for training below. All datasets contain English text. The table summarizes which emotions are available in each of the datasets. The datasets represent a diverse collection of text types. Specifically, they contain emotion labels for texts from Twitter, Reddit, student self-reports, and utterances from TV dialogues. As MELD (Multimodal EmotionLines Dataset) extends the popular EmotionLines dataset, EmotionLines itself is not included here.