File size: 340 Bytes
193bf75
 
 
 
 
 
 
 
 
 
e5924b8
8e99bfb
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="Falconsai/text_summarization")
        summary = summarizer(text, max_length=500, min_length=300, do_sample=False)
        return summary[0]['summary_text']