3a05chatgpt commited on
Commit
f540a3e
·
verified ·
1 Parent(s): ef6666a

Update textsumm.py

Browse files
Files changed (1) hide show
  1. textsumm.py +31 -11
textsumm.py CHANGED
@@ -1,11 +1,31 @@
1
- from transformers import pipeline
2
-
3
- # 建立中文摘要管道
4
- summarizer = pipeline("summarization", model="facebook/bart-large-cnn")
5
-
6
- def 文字摘要(輸入文本, max_length=130, min_length=30, do_sample=False):
7
- """
8
- 將輸入的純文字自動摘要為繁體中文重點
9
- """
10
- result = summarizer(輸入文本, max_length=max_length, min_length=min_length, do_sample=do_sample)
11
- return result
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
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(文字摘要(測試文本))