Update README.md

README.md

@@ -14,9 +14,10 @@ should probably proofread and complete it, then remove this comment. -->
 
 # mental_bert
 
-This model is a fine-tuned version of [mental/mental-bert-base-uncased](https://huggingface.co/mental/mental-bert-base-uncased) on an unknown dataset.
-It achieves the following results on the evaluation set:
-- Loss: 0.9179
+This model is a fine-tuned version of [mental/mental-bert-base-uncased](https://huggingface.co/mental/mental-bert-base-uncased) on [hackathon-somos-nlp-2023/DiagTrast](https://huggingface.co/datasets/hackathon-somos-nlp-2023/DiagTrast).
+It achieves the following results on the evaluation and test sets:
+- Evaluation Loss: 0.9179
+- Test Loss: 0.9831
 
 ## Model description