|
--- |
|
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"]) |
|
|
|
``` |
|
|