Update README.md
Browse files
README.md
CHANGED
@@ -11,9 +11,9 @@ If you want to know about training details or evaluation results, see [SlovakT5_
|
|
11 |
|
12 |
### How to use
|
13 |
```python
|
14 |
-
from transformers import
|
15 |
|
16 |
-
tokenizer =
|
17 |
model = T5ForConditionalGeneration.from_pretrained("ApoTro/slovak-t5-small")
|
18 |
|
19 |
input_ids = tokenizer("sst2 veta: Obraz je krajší, kvalitnejší a lepší.", return_tensors="pt").input_ids
|
|
|
11 |
|
12 |
### How to use
|
13 |
```python
|
14 |
+
from transformers import AutoTokenizer, T5ForConditionalGeneration
|
15 |
|
16 |
+
tokenizer = AutoTokenizer.from_pretrained("ApoTro/slovak-t5-small")
|
17 |
model = T5ForConditionalGeneration.from_pretrained("ApoTro/slovak-t5-small")
|
18 |
|
19 |
input_ids = tokenizer("sst2 veta: Obraz je krajší, kvalitnejší a lepší.", return_tensors="pt").input_ids
|