Bart Finetuned Conversational Summarization

🚀 文本摘要模型

本模型基于Facebook的BART（双向自回归变换器）架构，特别是为文本摘要任务进行微调的大型变体。BART是Facebook AI推出的一种序列到序列模型，能够处理包括摘要在内的各种自然语言处理任务。

🚀 快速开始

📦 安装指南

你可以使用pip安装必要的库：

pip install transformers

💻 使用示例

基础用法

以下是一个简单的代码片段，展示了如何在PyTorch中使用该模型进行段落摘要任务：

from transformers import AutoTokenizer, AutoModelForSeq2SeqLM

tokenizer = AutoTokenizer.from_pretrained("Mr-Vicky-01/Bart-Finetuned-conversational-summarization")
model = AutoModelForSeq2SeqLM.from_pretrained("Mr-Vicky-01/Bart-Finetuned-conversational-summarization")

def generate_summary(text):
    inputs = tokenizer([text], max_length=1024, return_tensors='pt', truncation=True)
    summary_ids = model.generate(inputs['input_ids'], max_new_tokens=100, do_sample=False)
    summary = tokenizer.decode(summary_ids[0], skip_special_tokens=True)
    return summary

text_to_summarize = """Now, there is no doubt that one of the most important aspects of any Pixel phone is its camera.
And there might be good news for all camera lovers. Rumours have suggested that the Pixel 9 could come with a telephoto lens,
improving its photography capabilities even further. Google will likely continue to focus on using AI to enhance its camera performance,
in order to make sure that Pixel phones remain top contenders in the world of mobile photography."""
summary = generate_summary(text_to_summarize)

print(summary)

高级用法

# 以下是模型训练参数的设置
num_train_epochs=1,
warmup_steps = 500,
per_device_train_batch_size=4,
per_device_eval_batch_size=4,
weight_decay = 0.01,
gradient_accumulation_steps=16

📚 详细文档

模型详情

空间链接

Summarization Model

如何微调此模型

Github

📄 许可证

本项目采用MIT许可证。