Harshb11 commited on
Commit
c2d5cdd
·
verified ·
1 Parent(s): 6cce2e6

Create text_summarization.py

Browse files
Files changed (1) hide show
  1. text_summarization.py +9 -0
text_summarization.py ADDED
@@ -0,0 +1,9 @@
 
 
 
 
 
 
 
 
 
 
1
+ from transformers import pipeline
2
+
3
+ class TextSummarizer:
4
+ def __init__(self):
5
+ self.summarizer = pipeline("summarization", model="sshleifer/distilbart-cnn-12-6")
6
+
7
+ def summarize(self, text):
8
+ result = self.summarizer(text, max_length=120, min_length=30, do_sample=False)
9
+ return result[0]['summary_text']