Commit
•
2b265d1
1
Parent(s):
f522bae
Fix example script (#1)
Browse files- Fix example script (ab9d6d87d4a10045ec488170c902740ca27b2d88)
Co-authored-by: Manuel Romero <mrm8488@users.noreply.huggingface.co>
README.md
CHANGED
@@ -73,7 +73,7 @@ model = T5ForConditionalGeneration.from_pretrained("grammarly/coedit-large")
|
|
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 |
|
|
|
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 |
|