machineteacher mrm8488 commited on
Commit
f5fd2e6
1 Parent(s): d1b1f9b

Fix example script (#1)

Browse files

- Fix example script (251d0ffc3e80314e576053c523ce8cec5c17a35c)


Co-authored-by: Manuel Romero <mrm8488@users.noreply.huggingface.co>

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