Spaces:
Sleeping
Sleeping
Update textsumm.py
Browse files- textsumm.py +31 -11
textsumm.py
CHANGED
@@ -1,11 +1,31 @@
|
|
1 |
-
|
2 |
-
|
3 |
-
|
4 |
-
|
5 |
-
|
6 |
-
|
7 |
-
|
8 |
-
|
9 |
-
|
10 |
-
|
11 |
-
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
1 |
+
# textsumm.py
|
2 |
+
# 中文化摘要模組,安裝 transformers 與 torch 即可使用
|
3 |
+
|
4 |
+
from transformers import pipeline
|
5 |
+
|
6 |
+
# 初始化摘要 pipeline
|
7 |
+
summarizer = pipeline("summarization", model="facebook/bart-large-cnn")
|
8 |
+
|
9 |
+
def 文字摘要(輸入文本, 最長長度=120, 最短長度=40):
|
10 |
+
"""
|
11 |
+
輸入:一段文本
|
12 |
+
輸出:摘要(中文說明)
|
13 |
+
"""
|
14 |
+
if len(輸入文本.strip()) == 0:
|
15 |
+
return "❗️ 請輸入需要摘要的內容。"
|
16 |
+
|
17 |
+
try:
|
18 |
+
結果 = summarizer(
|
19 |
+
輸入文本,
|
20 |
+
max_length=最長長度,
|
21 |
+
min_length=最短長度,
|
22 |
+
do_sample=False
|
23 |
+
)
|
24 |
+
return 結果[0]['summary_text']
|
25 |
+
except Exception as e:
|
26 |
+
return f"❌ 摘要生成失敗:{str(e)}"
|
27 |
+
|
28 |
+
# 若你要測試,可以取消下面註解
|
29 |
+
# if __name__ == "__main__":
|
30 |
+
# 測試文本 = "人工智慧(AI)是研究如何讓電腦模擬人類智能行為的學科,包括學習、推理、規劃、自然語言處理、知覺等。AI 技術已廣泛應用於語音辨識、影像分析、自駕車等領域。"
|
31 |
+
# print(文字摘要(測試文本))
|