Update README.md

README.md

@@ -1,11 +1,19 @@
 ---
-license: apache-2.0
+license: cc-by-4.0
 base_model: google/mt5-small
 tags:
 - generated_from_trainer
 model-index:
-- name: PII_MT5_SMALL
+- name: mt5-small-ai4privacy
   results: []
+datasets:
+- ai4privacy/pii-masking-65k
+language:
+- en
+- fr
+- de
+- it
+pipeline_tag: text2text-generation
 ---
 
 <!-- This model card has been generated automatically according to the information the Trainer had access to. You
@@ -13,7 +21,7 @@ should probably proofread and complete it, then remove this comment. -->
 
 # PII_MT5_SMALL
 
-This model is a fine-tuned version of [google/mt5-small](https://huggingface.co/google/mt5-small) on an unknown dataset.
+This model is a fine-tuned version of [google/mt5-small](https://huggingface.co/google/mt5-small) on [ai4privacy/pii-masking-65k](https://huggingface.co/datasets/ai4privacy/pii-masking-65k) dataset.
 
 ## Model description
 
@@ -50,4 +58,4 @@ The following hyperparameters were used during training:
 - Transformers 4.32.1
 - Pytorch 2.0.1
 - Datasets 2.14.4
-- Tokenizers 0.13.3
+- Tokenizers 0.13.3