Sadem-12 commited on
Commit
2e1c7d8
·
verified ·
1 Parent(s): 4e0e849

Create summarization.py

Browse files
Files changed (1) hide show
  1. summarization.py +10 -0
summarization.py ADDED
@@ -0,0 +1,10 @@
 
 
 
 
 
 
 
 
 
 
 
1
+ from transformers import pipeline
2
+
3
+ summarizer = pipeline("summarization", model="facebook/bart-large-cnn")
4
+
5
+ def summarize_text(text, level):
6
+ """تلخيص النص بمستويات مختلفة"""
7
+ length_map = {"مختصر": 30, "متوسط": 60, "تفصيلي": 100}
8
+ max_length = length_map.get(level, 60)
9
+ summary = summarizer(text, max_length=max_length, min_length=20, do_sample=False)
10
+ return summary[0]['summary_text']