Fix example script
#1
by
mrm8488
- opened
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)
|
| 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 |
|