Update README.md

README.md

@@ -6,14 +6,13 @@ metrics:
 model-index:
 - name: cantonese-chinese-translation-gen1
   results: []
+datasets:
+- raptorkwok/cantonese-chinese-dataset-gen2
 ---
 
-<!-- This model card has been generated automatically according to the information the Trainer had access to. You
-should probably proofread and complete it, then remove this comment. -->
+# Cantonese-Written Chinese Translation Model
 
-# cantonese-chinese-translation-gen1
-
-This model is a fine-tuned version of [fnlp/bart-base-chinese](https://huggingface.co/fnlp/bart-base-chinese) on an unknown dataset.
+This model is a fine-tuned version of [fnlp/bart-base-chinese](https://huggingface.co/fnlp/bart-base-chinese) on [Cantonese-Written Chinese Dataset Gen2](https://huggingface.co/raptorkwok/cantonese-chinese-dataset-gen2).
 It achieves the following results on the evaluation set:
 - Loss: 1.5413
 - Bleu: 40.7808
@@ -22,18 +21,20 @@ It achieves the following results on the evaluation set:
 
 ## Model description
 
-More information needed
+The model is based on BART Chinese model, trained on 1M Cantonese-Written Chinese Parallel Corpus data.
 
 ## Intended uses & limitations
 
-More information needed
+Its intended use is to translate Cantonese sentences to Written Chinese accurately.
 
 ## Training and evaluation data
 
-More information needed
+Training and evaluation data is provided by the [Cantonese-Written Chinese Dataset Gen2](https://huggingface.co/raptorkwok/cantonese-chinese-dataset-gen2).
 
 ## Training procedure
 
+The training was performed using `Seq2SeqTrainer`.
+
 ### Training hyperparameters
 
 The following hyperparameters were used during training:
@@ -62,4 +63,4 @@ The following hyperparameters were used during training:
 - Transformers 4.28.1
 - Pytorch 2.3.1+cu121
 - Datasets 2.19.1
-- Tokenizers 0.13.3
+- Tokenizers 0.13.3