metadata
license: apache-2.0
datasets:
- jfleg
widget:
- text: 'fix grammar: I am work with machine to write gooder english.'
example_title: example
This is my first model for grammar error correction. It uses the jfleg dataset and is built on t5-base. It is trained only for 3 epochs so the output isn't that great.
Usage
You can use this model with the standard transformers library. This model should be small enough to run on the CPU.
$ pip install transformers torch sentencepiece
Once you have the dependencies setup, you should be able to run this model.
from transformers import AutoModelForSeq2SeqLM, AutoTokenizer
model_name = 'vagmi/grammar-t5'
model = AutoModelForSeq2SeqLM.from_pretrained(model_name)
tokenizer = AutoTokenizer.from_pretrained(model_name)
text = 'fix grammar: I am work with machine to write gooder english.'
inputs = tokenizer(text, return_tensors='pt')
outputs = model.generate(inputs['input_ids'], num_beams=2, max_length=512, early_stopping=True)
fixed = tokenizer.decode(outputs[0], skip_special_tokens=True)
# I am working with machine to write better english.