File size: 1,154 Bytes
5d01bda
 
 
92c0981
 
5d01bda
92c0981
 
5d01bda
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
92c0981
5d01bda
 
 
 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
# textsumm.py
# 中文化摘要模組,安裝 transformers 與 torch 即可使用

from transformers import pipeline

# 初始化摘要 pipeline
summarizer = pipeline("summarization", model="facebook/bart-large-cnn")

def 文字摘要(輸入文本, 最長長度=120, 最短長度=40):
    """

    輸入:一段文本

    輸出:摘要(中文說明)

    """
    if len(輸入文本.strip()) == 0:
        return "❗️ 請輸入需要摘要的內容。"

    try:
        結果 = summarizer(
            輸入文本,
            max_length=最長長度,
            min_length=最短長度,
            do_sample=False
        )
        return 結果[0]['summary_text']
    except Exception as e:
        return f"❌ 摘要生成失敗:{str(e)}"

# 若你要測試,可以取消下面註解
# if __name__ == "__main__":
#     測試文本 = "人工智慧(AI)是研究如何讓電腦模擬人類智能行為的學科,包括學習、推理、規劃、自然語言處理、知覺等。AI 技術已廣泛應用於語音辨識、影像分析、自駕車等領域。"
#     print(文字摘要(測試文本))