Update README.md
Browse files
README.md
CHANGED
@@ -1,4 +1,6 @@
|
|
1 |
-
###
|
|
|
|
|
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 |
+
```
|