grammarly
/

coedit-xxl

Text2Text Generation Transformers PyTorch Safetensors

English t5 Inference Endpoints text-generation-inference

Model card Files Files and versions Community

mrm8488 commited on May 19, 2023

Commit

251d0ff

•

1 Parent(s): d1b1f9b

Fix example script

Files changed (1) hide show

README.md +1 -1

README.md CHANGED Viewed

@@ -73,7 +73,7 @@ model = T5ForConditionalGeneration.from_pretrained("grammarly/coedit-xxl")
 input_text = 'Fix grammatical errors in this sentence: New kinds of vehicles will be invented with new technology than today.'
 input_ids = tokenizer(input_text, return_tensors="pt").input_ids
 outputs = model.generate(input_ids, max_length=256)
-edited_text = tokenizer.decode(outputs[0], skip_special_tokens=True)[0]
 ```

 input_text = 'Fix grammatical errors in this sentence: New kinds of vehicles will be invented with new technology than today.'
 input_ids = tokenizer(input_text, return_tensors="pt").input_ids
 outputs = model.generate(input_ids, max_length=256)
+edited_text = tokenizer.decode(outputs[0], skip_special_tokens=True)
 ```