Spaces:
Sleeping
Sleeping
File size: 848 Bytes
fa33e09 54f5da8 fa33e09 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 |
from transformers import pipeline
class TextSummarizer:
def __init__(self):
print("Loading summarization model...")
self.summarizer = pipeline("summarization", model="facebook/bart-large-cnn")
print("Summarization model loaded successfully.")
def summarize_text(self, text, max_length=150, min_length=40):
# Check if the input text is too short for summarization
if len(text.split()) < 30:
print("Input text too short to summarize. Skipping summarization.")
return "Transcript too short to summarize."
print("Summarizing text...")
summary = self.summarizer(text, max_length=max_length, min_length=min_length, do_sample=False)
summarized_text = summary[0]['summary_text']
print("Text summarized successfully.")
return summarized_text
|