Update README.md

README.md

@@ -5,6 +5,12 @@ tags:
 model-index:
 - name: n3wtou/mt5-small-finedtuned-4-swahili
   results: []
+datasets:
+- csebuetnlp/xlsum
+language:
+- sw
+metrics:
+- rouge
 ---
 
 <!-- This model card has been generated automatically according to the information Keras had access to. You should
@@ -12,7 +18,7 @@ probably proofread and complete it, then remove this comment. -->
 
 # n3wtou/mt5-small-finedtuned-4-swahili
 
-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 the [csebuetnlp/xlsum](https://huggingface.co/datasets/csebuetnlp/xlsum/viewer/swahili/train) dataset.
 It achieves the following results on the evaluation set:
 - Train Loss: 3.0006
 - Validation Loss: 2.7015
@@ -20,7 +26,7 @@ It achieves the following results on the evaluation set:
 
 ## Model description
 
-More information needed
+This model is a fined-tuned google/mt5-small for Kiswahili abstractive text generation
 
 ## Intended uses & limitations
 
@@ -59,4 +65,4 @@ The following hyperparameters were used during training:
 - Transformers 4.29.2
 - TensorFlow 2.12.0
 - Datasets 2.12.0
-- Tokenizers 0.13.3
+- Tokenizers 0.13.3