Bart Finetuned Text Summarization

🚀 BART Large CNN Text Summarization Model

This model, based on the Facebook BART architecture, is fine - tuned for text summarization tasks, offering efficient solutions for condensing text.

🚀 Quick Start

This BART Large CNN model is designed for text summarization. It leverages the power of the BART architecture to generate accurate summaries.

✨ Features

• Powerful Architecture: Built on the BART Large CNN architecture, it offers high - performance text summarization.
• Fine - tuned: Specifically fine - tuned on the xsum dataset for optimal summarization results.

📦 Installation

You can install the necessary libraries using pip:

pip install transformers

💻 Usage Examples

Basic Usage

Here is a simple snippet of how to use this model for the task of paragraph summarization in PyTorch:

from transformers import AutoTokenizer, AutoModelForSeq2SeqLM

tokenizer = AutoTokenizer.from_pretrained("suriya7/bart-finetuned-text-summarization")
model = AutoModelForSeq2SeqLM.from_pretrained("suriya7/bart-finetuned-text-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)

Advanced Usage

The following is an example output of the model:

Google is rumoured to be about to unveil its next-generation Pixel smartphone,
the Google Pixel 9,which is expected to come with a telephoto lens and an artificial intelligence (AI)
system to improve its camera capabilities, as well as improve the quality of its images.

Training Parameters

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

📚 Documentation

Model Details

📄 License

This project is licensed under the MIT license.