Update README.md

README.md

@@ -71,23 +71,26 @@ emotion_labels = {
 print(f"Predicted emotion: {emotion_labels[emotion_id]}")
 
 ## Citation
-
-If you use this model, please cite both the model and original datasets:
-
+This model is fine-tuned on:
+
+- **EmoWOZ**: Dialogue emotion annotations
+  - Paper: [EmoWOZ: A Large-Scale Corpus for Emotion Recognition in Task-oriented Dialogues](https://arxiv.org/...)
+  - Citation:
+    ```bibtex
+    @dataset{emowoz2021,
+      title={EmoWOZ: A Large-Scale Corpus for Emotion Recognition in Task-oriented Dialogues},
+      author={...},
+      year={2021}
+    }
+    ```
+
+To cite this fine-tuned model, use:
 ```bibtex
-# Model citation
 @misc{emowoz-emotion-classifier,
   title={EmoWOZ Emotion Classifier},
-  author={Your Name},
+  author={joshthoo},
   year={2026},
   publisher={Hugging Face},
   howpublished={\url{https://huggingface.co/your-username/emowoz-roberta-emotion}}
 }
-
-# Dataset citation (find the official one and add here)
-@inproceedings{emowoz2021,
-  title={EmoWOZ: A Large-Scale Corpus for Emotion Recognition in Task-oriented Dialogues},
-  author={XXX},
-  booktitle={Proceedings of...},
-  year={2021}
-}
+```