Instructions to use sumedh/t5-base-amazonreviews with libraries, inference providers, notebooks, and local apps. Follow these links to get started.
- Libraries
- Transformers
How to use sumedh/t5-base-amazonreviews with Transformers:
# Use a pipeline as a high-level helper # Warning: Pipeline type "summarization" is no longer supported in transformers v5. # You must load the model directly (see below) or downgrade to v4.x with: # pip install "transformers<5.0.0" from transformers import pipeline pipe = pipeline("summarization", model="sumedh/t5-base-amazonreviews")# pip install -U transformers accelerate # Load model directly from transformers import AutoTokenizer, AutoModelForSeq2SeqLM tokenizer = AutoTokenizer.from_pretrained("sumedh/t5-base-amazonreviews") model = AutoModelForSeq2SeqLM.from_pretrained("sumedh/t5-base-amazonreviews", device_map="auto") - Notebooks
- Google Colab
- Kaggle
Merge branch 'main' of https://huggingface.co/sumedh/t5-base-amazonreviews
Browse files
README.md
CHANGED
|
@@ -1,3 +1,18 @@
|
|
| 1 |
---
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 2 |
license: apache-2.0
|
| 3 |
---
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
---
|
| 2 |
+
language:
|
| 3 |
+
- en
|
| 4 |
+
datasets:
|
| 5 |
+
- amazon_reviews_multi
|
| 6 |
+
tags:
|
| 7 |
+
- summarization
|
| 8 |
+
|
| 9 |
license: apache-2.0
|
| 10 |
---
|
| 11 |
+
|
| 12 |
+
T5-base model for text summarization finetuned on subset of amazon reviews for english language.
|
| 13 |
+
|
| 14 |
+
## Rouge scores
|
| 15 |
+
- Rouge 1 : 0.5019
|
| 16 |
+
- Rouge 2 : 0.4226
|
| 17 |
+
- Rouge L : 0.4877
|
| 18 |
+
- Rouge Lsum : 0.4877
|