Kaveh commited on
Commit
f0787fe
·
unverified ·
1 Parent(s): 3182a34

Create summarize.py

Browse files
Files changed (1) hide show
  1. summarizer/summarize.py +17 -0
summarizer/summarize.py ADDED
@@ -0,0 +1,17 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ from summarizer.models import load_model
2
+
3
+ # مدل رو فقط یک‌بار بارگذاری می‌کنیم
4
+ model = load_model()
5
+
6
+ def summarize_text(text: str) -> str:
7
+ if not text.strip():
8
+ return "متنی برای خلاصه‌سازی وجود ندارد."
9
+
10
+ # تقسیم به پاراگراف‌های کوچکتر اگه خیلی بزرگ باشه
11
+ if len(text) > 2000:
12
+ text = text[:2000]
13
+
14
+ try:
15
+ return model(text)
16
+ except Exception as e:
17
+ return f"در خلاصه‌سازی خطا رخ داد: {str(e)}"