Update README.md

README.md

@@ -84,19 +84,14 @@ The following hyperparameters were used during training:
 
 ### How to use model
 ```py
-from transformers import pipeline
-from random import randrange
- 
+from transformers import AutoTokenizer, AutoModelForSeq2SeqLM, pipeline
 
-summarizer = pipeline("summarization", model="sharmax-vikas/flan-t5-base-samsum", device=0)
- 
+ckpt = 'sharmax-vikas/flan-t5-base-samsum'
+tokenizer = AutoTokenizer.from_pretrained(ckpt)
+# Use AutoModelForSeq2SeqLM for text generation tasks like summarization
+model = AutoModelForSeq2SeqLM.from_pretrained(ckpt) 
 
-
-print(f"dialogue: \n{sample['dialogue']}\n---------------")
-
-res = summarizer(sample["dialogue"])
- 
-print(f"flan-t5-base summary:\n{res[0]['summary_text']}")
+summarize = pipeline('summarization', tokenizer=tokenizer, model=model)
 
 result = summarize('''Hannah: Hey, do you have Betty's number?
 Amanda: Lemme check