README.md

metadata

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

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