machineteacher commited on
Commit
7091dbd
1 Parent(s): 2d14f20

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +2 -2
README.md CHANGED
@@ -67,8 +67,8 @@ Given an edit instruction and an original text, our model can generate the edite
67
  ```python
68
  from transformers import AutoTokenizer, T5ForConditionalGeneration
69
 
70
- tokenizer = AutoTokenizer.from_pretrained("grammarly/coedit-large")
71
- model = T5ForConditionalGeneration.from_pretrained("grammarly/coedit-large")
72
  input_text = 'Fix grammatical errors in this sentence: New kinds of vehicles will be invented with new technology than today.'
73
  input_ids = tokenizer(input_text, return_tensors="pt").input_ids
74
  outputs = model.generate(input_ids, max_length=256)
 
67
  ```python
68
  from transformers import AutoTokenizer, T5ForConditionalGeneration
69
 
70
+ tokenizer = AutoTokenizer.from_pretrained("grammarly/coedit-xl")
71
+ model = T5ForConditionalGeneration.from_pretrained("grammarly/coedit-xl")
72
  input_text = 'Fix grammatical errors in this sentence: New kinds of vehicles will be invented with new technology than today.'
73
  input_ids = tokenizer(input_text, return_tensors="pt").input_ids
74
  outputs = model.generate(input_ids, max_length=256)