Update README.md

README.md

@@ -19,12 +19,11 @@ probably proofread and complete it, then remove this comment. -->
 
 # TF-Finetuned-xsum
 
-This model is a fine-tuned version of [t5-small](https://huggingface.co/t5-small) on an unknown dataset.
+This model is a fine-tuned version of [t5-small](https://huggingface.co/t5-small) on [xsum](https://huggingface.co/datasets/xsum) dataset.
 It achieves the following results on the evaluation set:
 - Train Loss: 2.8326
 - Validation Loss: 2.5943
-- Train Rougel: tf.Tensor(0.1999889, shape=(), dtype=float32)
-- Epoch: 0
+- Epoch: 1
 
 ## Model description
 
@@ -50,7 +49,7 @@ The following hyperparameters were used during training:
 
 | Train Loss | Validation Loss | Train Rougel                                  | Epoch |
 |:----------:|:---------------:|:---------------------------------------------:|:-----:|
-| 2.8326     | 2.5943          | tf.Tensor(0.1999889, shape=(), dtype=float32) | 0     |
+| 2.8326     | 2.5943          | tf.Tensor(0.1999889, shape=(), dtype=float32) | 1     |
 
 
 ### Framework versions