Downloads · 30 days
324K
2% of all-time downloads
t5-base-grammar-correction
vennify/t5-base-grammar-correction
t5-base-grammar-correction is a machine learning model from vennify. Use it for the machine learning task on the model card, and read the license before you ship it in a product. It is set up for transformers. The card lists the license as cc-by-nc-sa-4.0.
This model generates a revised version of inputted text with the goal of containing fewer grammatical errors. It was trained with Happy Transformer using a dataset called JFLEG. Here's a full article on how to train a…
All-time downloads
20.8M
Public
Repo size
1.8 GB
Likes
183
Public
Repo makeup
Click a slice to open those files.
.bin892 MB · 100%
At a glance
- Library
- transformers
- License
- cc-by-nc-sa-4.0
- Model type
- t5
- Access
- Public
- Created
- Mar 2, 2022
- Updated
- Jan 14, 2022
- SHA
- 9e4a09d2
Datasets
Model card
From the Hugging Face model README
T5 Grammar Correction
This model generates a revised version of inputted text with the goal of containing fewer grammatical errors. It was trained with Happy Transformer using a dataset called JFLEG. Here's a full article on how to train a similar model.
Usage
pip install happytransformer
from happytransformer import HappyTextToText, TTSettings
happy_tt = HappyTextToText("T5", "vennify/t5-base-grammar-correction")
args = TTSettings(num_beams=5, min_length=1)
# Add the prefix "grammar: " before each input
result = happy_tt.generate_text("grammar: This sentences has has bads grammar.", args=args)
print(result.text) # This sentence has bad grammar.
- Library
- transformers
- Type
- t5
- License
- cc-by-nc-sa-4.0
- Languages
- en
- Created
- Mar 2, 2022
- Updated
- Jan 14, 2022