# Load model directly
from transformers import AutoTokenizer, AutoModelForSeq2SeqLM
tokenizer = AutoTokenizer.from_pretrained("neuronstarml/samsuntextsum")
model = AutoModelForSeq2SeqLM.from_pretrained("neuronstarml/samsuntextsum")Quick Links
Model obtained by Fine Tuning 'google/pegasus-cnn_dailymail' using SAMSUM Dataset!
Usage
Example 1
from transformers import pipeline
summarizer = pipeline("summarization", model="neuronstarml/samsuntextsum")
text = '''
Hannah: Hey, do you have Betty's number?
Amanda: Lemme check
Hannah: <file_gif>
Amanda: Sorry, can't find it.
Amanda: Ask Larry
Amanda: He called her last time we were at the park together
Hannah: I don't know him well
Hannah: <file_gif>
Amanda: Don't be shy, he's very nice
Hannah: If you say so..
Hannah: I'd rather you texted him
Amanda: Just text him 🙂
Hannah: Urgh.. Alright
Hannah: Bye
Amanda: Bye bye
'''
summarizer(text)
- Downloads last month
- 7
# Use a pipeline as a high-level helper # Warning: Pipeline type "summarization" is no longer supported in transformers v5. # You must load the model directly (see below) or downgrade to v4.x with: # 'pip install "transformers<5.0.0' from transformers import pipeline pipe = pipeline("summarization", model="neuronstarml/samsuntextsum")