mrm8488 commited on
Commit
2f4b5e5
1 Parent(s): 2bf41f1

Fix example script

Browse files
Files changed (1) hide show
  1. README.md +1 -1
README.md CHANGED
@@ -76,7 +76,7 @@ model = T5ForConditionalGeneration.from_pretrained("grammarly/coedit-xl-composit
76
  input_text = 'Fix grammatical errors in this sentence and make it simpler: New kinds of vehicles will be invented with new technology than today.'
77
  input_ids = tokenizer(input_text, return_tensors="pt").input_ids
78
  outputs = model.generate(input_ids, max_length=256)
79
- edited_text = tokenizer.decode(outputs[0], skip_special_tokens=True)[0]
80
  ```
81
 
82
 
 
76
  input_text = 'Fix grammatical errors in this sentence and make it simpler: New kinds of vehicles will be invented with new technology than today.'
77
  input_ids = tokenizer(input_text, return_tensors="pt").input_ids
78
  outputs = model.generate(input_ids, max_length=256)
79
+ edited_text = tokenizer.decode(outputs[0], skip_special_tokens=True)
80
  ```
81
 
82