transformers safetensors bart text2text-generation summarization en dataset:EdinburghNLP/xsum license:mit autotrain_compatible endpoints_compatible region:us
---
datasets:
- EdinburghNLP/xsum
language:
- en
license: mit
pipeline_tag: summarization
widget:
- text: 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.
- text: The Samastha Kerala Sunni Students Federation (SKSSF) has also expressed concern
    over holding the election on Friday. In a statement issued in Kozhikode on Saturday,
    SKSSF state secretariat asked the EC to postpone the election to another day.
    It said conducting elections on Friday will cause inconvenience to people from
    the Muslim community deputed on poll duty or as booth agents of political parties
    to participate in Friday juma prayers. Meanwhile, the Wisdom Islamic Organisation
    has asked the state government to officially demand the EC to hold the elections
    in Kerala and Tamil Nadu on some other day, citing inconvenience of believers.
    State president P N Abdul Latheef Madani said all secular forces should put pressure
    on the poll panel to change the date of elections.
---

# BART Large CNN Text Summarization Model

This model is based on the Facebook BART (Bidirectional and Auto-Regressive Transformers) architecture, specifically the large variant fine-tuned for text summarization tasks. BART is a sequence-to-sequence model introduced by Facebook AI, capable of handling various natural language processing tasks, including summarization.

## Model Details:

- **Architecture**: BART Large CNN
- **Pre-trained model**: BART Large
- **Fine-tuned for**: Text Summarization
- **Fine-tuning dataset**: [xsum](https://huggingface.co/datasets/EdinburghNLP/xsum)

## Usage:

### Installation:

You can install the necessary libraries using pip:
```bash
pip install transformers
```
### Inferecnce
provided a simple snippet of how to use this model for the task of paragraph summarization in PyTorch.
```python
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)
```

```
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
```python
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
```