Update README.md

README.md

@@ -24,8 +24,13 @@ should probably proofread and complete it, then remove this comment. -->
 
 # bert-finetuned-bpmn
 
-This model is a fine-tuned version of [bert-base-cased](https://huggingface.co/bert-base-cased) on an unknown dataset.
-It achieves the following results on the evaluation set:
+This model is a fine-tuned version of [bert-base-cased](https://huggingface.co/bert-base-cased) on a dataset containing the following labels:
+* `AGENT`
+* `TASK`
+
+The dataset (and the notebook used for training) can be found on the following GitHub repo: https://github.com/jtlicardo/bert-finetuned-bpmn
+
+The model achieves the following results on the evaluation set:
 - Loss: 0.2656
 - Precision: 0.7314
 - Recall: 0.8366