Update README.md

README.md

@@ -5,6 +5,13 @@ tags:
 model-index:
 - name: clinical-t5
   results: []
+datasets:
+- AGBonnet/augmented-clinical-notes
+language:
+- en
+metrics:
+- rouge
+pipeline_tag: summarization
 ---
 
 <!-- This model card has been generated automatically according to the information the Trainer had access to. You
@@ -12,21 +19,13 @@ should probably proofread and complete it, then remove this comment. -->
 
 # clinical-t5
 
-This model is a fine-tuned version of [google-t5/t5-small](https://huggingface.co/google-t5/t5-small) on an unknown dataset.
-
-## Model description
-
-More information needed
+This is a finetuned T5-small model from Google, a checkpoint with 60 million parameters, for clinical note summarization.
+It was fine-tuned with the [augmented-clinical-notes](https://huggingface.co/datasets/AGBonnet/augmented-clinical-notes) dataset, available in the Hugging Face. 
 
 ## Intended uses & limitations
 
-More information needed
-
-## Training and evaluation data
-
-More information needed
-
-## Training procedure
+The model was created for learning purposes. Hence, although being briefly evaluated in [this](https://github.com/hossboll/clinical_nlp/blob/main/clinical_t5_finetuned.ipynb
+) notebook, it should be further refined.
 
 ### Training hyperparameters
 
@@ -44,4 +43,4 @@ The following hyperparameters were used during training:
 - Transformers 4.30.0
 - Pytorch 2.2.1+cu121
 - Datasets 2.19.1
-- Tokenizers 0.13.3
+- Tokenizers 0.13.3