File size: 335 Bytes
193bf75
 
 
 
 
 
 
 
 
 
 
 
 
1
2
3
4
5
6
7
8
9
10
11
12
13
from transformers import pipeline



class Summarizer:

    def __init__(self):
        pass

    def summarize_text(self, text):
        summarizer = pipeline("summarization", model="facebook/bart-large-cnn")
        summary = summarizer(text, max_length=500, min_length=100, do_sample=False)
        return summary[0]['summary_text']