ApoTro commited on
Commit
40724f2
1 Parent(s): a87367e

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +2 -2
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 T5Tokenizer, T5ForConditionalGeneration
15
 
16
- tokenizer = T5Tokenizer.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
 
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