addy88 commited on
Commit
c33cdd8
1 Parent(s): 66c3f8b

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +6 -2
README.md CHANGED
@@ -1,4 +1,6 @@
1
- ### Python
 
 
2
  from transformers import AutoTokenizer, AutoModelForSeq2SeqLM
3
 
4
  tokenizer = AutoTokenizer.from_pretrained("addy88/t5-grammar-correction")
@@ -9,4 +11,6 @@ input_ids = tokenizer('grammar: This sentences has has bads grammar.', return_te
9
 
10
  outputs = model.generate(input_ids)
11
 
12
- print(tokenizer.decode(outputs[0], skip_special_tokens=True))
 
 
1
+ ### How to use
2
+ Here is how to use this model in PyTorch:
3
+ ```python
4
  from transformers import AutoTokenizer, AutoModelForSeq2SeqLM
5
 
6
  tokenizer = AutoTokenizer.from_pretrained("addy88/t5-grammar-correction")
11
 
12
  outputs = model.generate(input_ids)
13
 
14
+ print(tokenizer.decode(outputs[0], skip_special_tokens=True))
15
+
16
+ ```