Update README.md

README.md

@@ -9,6 +9,10 @@ base_model: bert-base-uncased
 model-index:
 - name: final-lr2e-5-bs16-fp16-2
   results: []
+language:
+- en
+library_name: transformers
+pipeline_tag: text-classification
 ---
 
 <!-- This model card has been generated automatically according to the information the Trainer had access to. You
@@ -16,7 +20,7 @@ should probably proofread and complete it, then remove this comment. -->
 
 # final-lr2e-5-bs16-fp16-2
 
-This model is a fine-tuned version of [bert-base-uncased](https://huggingface.co/bert-base-uncased) on an unknown dataset.
+This model is a fine-tuned version of [bert-base-uncased](https://huggingface.co/bert-base-uncased) on an https://github.com/rewire-online/edos dataset.
 It achieves the following results on the evaluation set:
 - Loss: 0.4823
 - F1 Macro: 0.8301
@@ -95,4 +99,4 @@ weighted avg   0.876269  0.879250  0.877172  4000.00000 |
 - Transformers 4.27.0.dev0
 - Pytorch 1.13.1+cu117
 - Datasets 2.9.0
-- Tokenizers 0.13.2
+- Tokenizers 0.13.2