Update README.md

README.md

@@ -23,7 +23,8 @@ metrics:
  <a href="https://colab.research.google.com/drive/1fNOfy7oHYETI_KzJSz8JrhYohFBBl0HY">
   <img src="https://colab.research.google.com/assets/colab-badge.svg" alt="Open In Colab"/>
 </a>
-This model is a further trained version of [jordiclive/flan-t5-11b-summarizer-filtered](https://huggingface.co/jordiclive/flan-t5-11b-summarizer-filtered).
+
+**Note** This model is a further trained version of [jordiclive/flan-t5-11b-summarizer-filtered](https://huggingface.co/jordiclive/flan-t5-11b-summarizer-filtered).
 
 
 A fine-tuned version of [google/flan-t5-xxl](https://huggingface.co/google/flan-t5-xxl) on various summarization datasets (xsum, wikihow, cnn_dailymail/3.0.0, samsum, scitldr/AIC, billsum, TLDR, wikipedia-summary)
@@ -59,7 +60,7 @@ This pipeline will run slower and not have some of the tokenization parameters a
 ```python
 from transformers import pipeline
 
-summarizer = pipeline("summarization", "jordiclive/flan-t5-11b-summarizer-filtered", torch_dtype=torch.bfloat16)
+summarizer = pipeline("summarization", "jordiclive/flan-t5-11b-summarizer-filtered-1.5-epoch", torch_dtype=torch.bfloat16)
 
 raw_document = 'You must be 18 years old to live or work in New York State...'
 prompt = "Summarize the following article in 10-20 words:"