AI_Study_Assistant / summarization.py
Sadem-12's picture
Create summarization.py
2e1c7d8 verified
raw
history blame
442 Bytes
from transformers import pipeline
summarizer = pipeline("summarization", model="facebook/bart-large-cnn")
def summarize_text(text, level):
"""تلخيص النص بمستويات مختلفة"""
length_map = {"مختصر": 30, "متوسط": 60, "تفصيلي": 100}
max_length = length_map.get(level, 60)
summary = summarizer(text, max_length=max_length, min_length=20, do_sample=False)
return summary[0]['summary_text']