Updated with code

This commit is contained in:
DataAnts-AI
2025-01-28 17:00:03 -05:00
parent 2b37a68c56
commit 3c44257d8a
8 changed files with 180 additions and 0 deletions

8
utils/summarization.py Normal file
View File

@ -0,0 +1,8 @@
from transformers import pipeline
SUMMARY_MODEL = "Falconsai/text_summarization"
def summarize_text(text):
"""Summarize text using a Hugging Face pipeline."""
summarizer = pipeline("summarization", model=SUMMARY_MODEL)
return summarizer(text, max_length=150, min_length=30, do_sample=False)[0]["summary_text"]