数据集:

• Samsung/samsum 基础模型:
• google-t5/t5-small 任务标签: 文本摘要

基于T5 Small模型的对话摘要系统

使用方法

from transformers import AutoModelForSeq2SeqLM, AutoTokenizer

模型检查点 = "ahlad/t5-small-finetuned-samsum"
分词器 = AutoTokenizer.from_pretrained(模型检查点)
模型 = AutoModelForSeq2SeqLM.from_pretrained(模型检查点)

输入文本 = """
艾玛：我借你的书看完了吗？
利亚姆：看完了，简直爱不释手！结局的反转太震撼了。
艾玛：是吧？我完全没想到。你觉得主角塑造得怎么样？
利亚姆：说实话刚开始觉得有点烦人，但后来慢慢喜欢上了。
艾玛：我也是。特别喜欢角色后期的成长。想看这个系列的下一本吗？
利亚姆：当然！快拿给我吧。
"""

输入参数 = 分词器(输入文本, return_tensors="pt")

输出结果 = 模型.generate(**输入参数)
摘要内容 = 分词器.decode(输出结果[0], skip_special_tokens=True)

print("摘要:", 摘要内容)