gaur3009 commited on
Commit
72e4742
Β·
verified Β·
1 Parent(s): 30b1890

Create summarizer.py

Browse files
Files changed (1) hide show
  1. summarizer.py +8 -0
summarizer.py ADDED
@@ -0,0 +1,8 @@
 
 
 
 
 
 
 
 
 
1
+ from transformers import pipeline
2
+
3
+ # load once
4
+ summarizer = pipeline("summarization", model="facebook/bart-large-cnn")
5
+
6
+ def summarize_text(text, max_length=100):
7
+ """Summarize long text into shorter summary."""
8
+ return summarizer(text, max_length=max_length, min_length=30, do_sample=False)[0]['summary_text']