File size: 716 Bytes
9cd6155 60df5f2 638f700 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 |
---
language: en
tags:
- grammar
- text2text-generation
datasets:
- leslyarun/c4_200m_gec_train100k_test25k
---
# Get Grammatical corrections on your English text, trained on a subset of c4-200m dataset
# Use the below code for running the model
``` python
from transformers import AutoTokenizer, AutoModelForSeq2SeqLM
from transformers import pipeline
tokenizer = AutoTokenizer.from_pretrained("leslyarun/grammatical-error-correction")
model = AutoModelForSeq2SeqLM.from_pretrained("leslyarun/grammatical-error-correction")
text2text_generator = pipeline("text2text-generation", model=model, tokenizer=tokenizer)
output = text2text_generator("grammar: " + sentence)
print(output[0]["generated_text"])
```
|