Spaces:
Sleeping
Sleeping
# 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(文字摘要(測試文本)) | |