YAML Metadata Warning:empty or missing yaml metadata in repo card
Check out the documentation for more information.
🟨 3. Review Summarization(FLAN-T5)
# Review Summarization (FLAN-T5)
## Overview
This model generates concise summaries from customer reviews.
It helps transform long reviews into short, meaningful insights.
The model is based on FLAN-T5 and fine-tuned for text summarization tasks.
## Model Details
- Base model: FLAN-T5
- Task: Text Summarization (Text2Text Generation)
## Dataset
Dataset used:
- Amazon Polarity Dataset
A subset of reviews was used for training summarization.
## Evaluation Results
| Model | ROUGE-1 | ROUGE-2 | ROUGE-L |
|--------|---------|---------|---------|
| FLAN-T5 | 0.106 | 0.021 | 0.096 |
## Usage
```python
from transformers import AutoTokenizer, AutoModelForSeq2SeqLM
model_name = "JerryJJJJJ/review-summarization-flan-t5"
tokenizer = AutoTokenizer.from_pretrained(model_name)
model = AutoModelForSeq2SeqLM.from_pretrained(model_name)
text = "The phone has great performance but poor battery life."
inputs = tokenizer(text, return_tensors="pt", truncation=True)
outputs = model.generate(**inputs, max_length=30)
summary = tokenizer.decode(outputs[0], skip_special_tokens=True)
print(summary)
- Downloads last month
- 5
Inference Providers NEW
This model isn't deployed by any Inference Provider. 🙋 Ask for provider support