Update README.md

README.md

@@ -2,17 +2,23 @@
 tags:
 - generated_from_keras_callback
 model-index:
-- name: pegasus-indonesian-base_finetune
+- name: pegasus_indonesian_base-finetune
   results: []
+license: apache-2.0
+datasets:
+- csebuetnlp/xlsum
+language:
+- id
+metrics:
+- rouge
+pipeline_tag: summarization
 ---
 
-
-
-# pegasus-indonesian-base_finetune
+# pegasus_indonesian_base-finetune
 
 Github : [PegasusAnthony](https://github.com/nicholaswilven/PEGASUSAnthony/tree/master)
 
-This model is a fine-tuned version of [pegasus-indonesian-base_pretrained](https://huggingface.co/thonyyy/pegasus-indonesian-base_pretrained) on [Indosum](https://paperswithcode.com/dataset/indosum), [Liputan6](https://paperswithcode.com/dataset/liputan6) and [XLSum](https://huggingface.co/datasets/csebuetnlp/xlsum).
+This model is a fine-tuned version of [pegasus_indonesian_base-pretrained](https://huggingface.co/thonyyy/pegasus_indonesian_base-pretrained) on [Indosum](https://paperswithcode.com/dataset/indosum), [Liputan6](https://paperswithcode.com/dataset/liputan6) and [XLSum](https://huggingface.co/datasets/csebuetnlp/xlsum).
 
 It achieves the following results on the evaluation set:
 - Train Loss: 1.6196
@@ -45,7 +51,7 @@ Finetune dataset:
 ```python
 # Load model and tokenizer
 from transformers import TFPegasusForConditionalGeneration, PegasusTokenizerFast
-model_name = "thonyyy/pegasus-indonesian-base_finetune"
+model_name = "thonyyy/pegasus_indonesian_base-finetune"
 model = TFPegasusForConditionalGeneration.from_pretrained(model_name)
 tokenizer = PegasusTokenizerFast.from_pretrained(model_name)
 
@@ -103,4 +109,4 @@ The following hyperparameters were used during training:
 - Tokenizers 0.13.3
 
 ### Special Thanks
-Research supported with Cloud TPUs from Google’s TPU Research Cloud (TRC)
+Research supported with Cloud TPUs from Google’s TPU Research Cloud (TRC)