Update README.md

README.md

@@ -15,39 +15,71 @@ tags:
 
 [![Generic badge](https://img.shields.io/badge/STATUS-WIP-yellow.svg)](https://shields.io/)
 
->  ⚠️ ATTENTION
-> 
-> NOTE THAT FOR THE MODEL TO WORK AS INTENDED, YOU NEED TO APPEND THE 'summarize:' PREFIX BEFORE THE INPUT DATA
+[![Open in Collab](https://colab.research.google.com/assets/colab-badge.svg)]()
 
 # Table of Contents
 
 1. [Model Details](#model-details)
-2. [Uses](#uses)
+2. [Usage](#usage)
 3. [Training Details](#training-details)
-4. [Evaluation](#evaluation)
-5. [How To Get Started With the Model](#how-to-get-started-with-the-model)
-6. [Citation](#citation)
-7. [Author](#model-card-authors)
+4. [Training Results](#training-results)
+5. [Citation](#citation)
+6. [Author](#model-card-authors)
 
 # Model Details
 
-This T5 model, named `KipperDev/t5_summarizer_model`, is fine-tuned specifically for the task of document summarization. It's based on the T5 architecture, renowned for its flexibility and efficiency across a wide range of NLP tasks, including summarization. This model aims to generate concise, coherent, and informative summaries from extensive text documents, leveraging the power of the T5's text-to-text approach.
+This variant of the [t5-small](https://huggingface.co/google-t5/t5-small) model, is fine-tuned specifically for the task of text summarization. This model aims to generate concise, coherent, and informative summaries from extensive text documents, leveraging the power of the T5's text-to-text approach.
 
-# Uses
+# Usage
 
-This model is intended for use in summarizing long-form documents into concise, informative abstracts. It's particularly useful for professionals and researchers who need to quickly grasp the essence of detailed reports, research papers, or articles without reading the entire text. 
+This model is intended for use in summarizing long-form texts into concise, informative abstracts. It's particularly useful for professionals and researchers who need to quickly grasp the essence of detailed reports, research papers, or articles without reading the entire text. 
+
+## Get Started
+
+Install with `pip`:
+
+```bash
+pip install transformers
+```
+
+Use in python:
+
+```python
+from transformers import pipeline
+from transformers import AutoTokenizer
+from transformers import AutoModelForSeq2SeqLM
+
+model_name = "KipperDev/t5_summarizer_model"
+
+tokenizer = AutoTokenizer.from_pretrained(model_name)
+model = AutoModelForSeq2SeqLM.from_pretrained(model_name)
+summarizer = pipeline("summarization", model=model, tokenizer=tokenizer)
+
+# Example usage
+prefix = "summarize: "
+input_text = "Your input text here."
+input_ids = tokenizer.encode(prefix + input_text, return_tensors="pt")
+summary_ids = model.generate(input_ids)
+summary = tokenizer.decode(summary_ids[0], skip_special_tokens=True)
+
+print(summary)
+```
+
+**NOTE THAT FOR THE MODEL TO WORK AS INTENDED, YOU NEED TO APPEND THE 'summarize:' PREFIX BEFORE THE INPUT DATA**
 
 # Training Details
 
 ## Training Data
 
-The model was trained using the [Big Patent Dataset](https://huggingface.co/datasets/big_patent), comprising 1.3 million US patent documents and their corresponding human-written summaries. This dataset was chosen for its rich language and complex structure, representative of the challenging nature of document summarization tasks. Training involved multiple subsets of the dataset to ensure broad coverage and robust model performance across varied document types.
+The model was trained using the [Big Patent Dataset](https://huggingface.co/datasets/big_patent), comprising 1.3 million US patent documents and their corresponding human-written summaries. This dataset was chosen for its rich language and complex structure, representative of the challenging nature of document summarization tasks. 
+
+Training involved multiple subsets of the dataset to ensure broad coverage and robust model performance across varied document types.
 
 ## Training Procedure
 
-Training was conducted over three rounds, with initial settings including a learning rate of 0.00002, batch size of 8, and 4 epochs. Subsequent rounds adjusted these parameters to refine model performance further. A linear decay learning rate schedule was applied to enhance model learning efficiency over time.
+Training was conducted over three rounds, with initial settings including a learning rate of 0.00002, batch size of 8, and 4 epochs. Subsequent rounds adjusted these parameters to refine model performance further, for respectively 0.0003, 8 and 12. As well, a linear decay learning rate schedule was applied to enhance model learning efficiency over time.
 
-# Evaluation
+# Training results
 
 Model performance was evaluated using the ROUGE metric, highlighting its capability to generate summaries closely aligned with human-written abstracts.
 
@@ -64,29 +96,6 @@ Model performance was evaluated using the ROUGE metric, highlighting its capabil
 | Steps per Second                        | 0.336     |
 
 
-# How to Get Started with the Model
-
-Use the code below to get started with the model.
-
-<details>
-<summary> Click to expand </summary>
-
-```python
-from transformers import T5Tokenizer, T5ForConditionalGeneration
-
-tokenizer = T5Tokenizer.from_pretrained("KipperDev/t5_summarizer_model")
-model = T5ForConditionalGeneration.from_pretrained("KipperDev/t5_summarizer_model")
-
-# Example usage
-prefix = "summarize: "
-input_text = "Your input text here."
-input_ids = tokenizer.encode(prefix + input_text, return_tensors="pt")
-summary_ids = model.generate(input_ids)
-summary = tokenizer.decode(summary_ids[0], skip_special_tokens=True)
-
-print(summary)
-```
-
 # Citation
 
 **BibTeX:**