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