pdf-tools-suite / textsumm.py
3a05chatgpt's picture
Upload 8 files
5d01bda verified
raw
history blame
1.15 kB
# 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(文字摘要(測試文本))