SCBconsulting commited on
Commit
2c2937b
·
verified ·
1 Parent(s): 021dc25

Create summarizer.py

Browse files
Files changed (1) hide show
  1. utils/summarizer.py +11 -0
utils/summarizer.py ADDED
@@ -0,0 +1,11 @@
 
 
 
 
 
 
 
 
 
 
 
 
1
+ from transformers import pipeline
2
+
3
+ summarizer = pipeline("summarization", model="facebook/bart-large-cnn")
4
+
5
+ def summarize_text(text):
6
+ if not text.strip():
7
+ return "No input provided."
8
+
9
+ text = text.replace("\n", " ").strip()[:1024]
10
+ summary = summarizer(text, max_length=130, min_length=30, do_sample=False)
11
+ return summary[0]["summary_text"]