Update README.md

README.md

@@ -7,19 +7,15 @@ metrics:
 model-index:
 - name: lsg-bart-large-4096-booksum
   results: []
+datasets:
+- ubaada/booksum-complete-cleaned
 ---
 
-<!-- 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. -->
 
 # lsg-bart-large-4096-booksum
 
-This model is a fine-tuned version of [ubaada/lsg-bart-large-4096-booksum](https://huggingface.co/ubaada/lsg-bart-large-4096-booksum) on an unknown dataset.
-It achieves the following results on the evaluation set:
-- Loss: 2.0742
-- Rouge1: 0.4145
-- Rouge2: 0.0797
-- Rougel: 0.1541
+This model is a fine-tuned version of [ubaada/lsg-bart-large-4096-booksum](https://huggingface.co/ubaada/lsg-bart-large-4096-booksum) on an ubaada/booksum-complete-cleaned dataset.
+Validation Loss (Subset of validation dataset) Loss: 2.0742
 
 ## Model description
 
@@ -49,18 +45,10 @@ The following hyperparameters were used during training:
 - num_epochs: 3
 - mixed_precision_training: Native AMP
 
-### Training results
-
-| Training Loss | Epoch | Step | Validation Loss | Rouge1 | Rouge2 | Rougel |
-|:-------------:|:-----:|:----:|:---------------:|:------:|:------:|:------:|
-| 1.3801        | 1.0   | 1251 | 2.0441          | 0.4223 | 0.0811 | 0.1532 |
-| 1.2385        | 2.0   | 2502 | 2.0753          | 0.3995 | 0.0751 | 0.1512 |
-| 0.9542        | 3.0   | 3753 | 2.0742          | 0.4145 | 0.0797 | 0.1541 |
-
 
 ### Framework versions
 
 - Transformers 4.40.2
 - Pytorch 2.2.0
 - Datasets 2.19.1
-- Tokenizers 0.19.1
+- Tokenizers 0.19.1