Pavol Balážik commited on
Commit
9188929
·
1 Parent(s): 5736520

Update README.md - wrong tensorflow model path

Browse files
Files changed (1) hide show
  1. README.md +2 -2
README.md CHANGED
@@ -61,8 +61,8 @@ output = model(**encoded_input)
61
  and in TensorFlow:
62
  ```python
63
  from transformers import RobertaTokenizer, TFRobertaModel
64
- tokenizer = RobertaTokenizer.from_pretrained('roberta-base')
65
- model = TFRobertaModel.from_pretrained('roberta-base')
66
  text = "Text ktorý sa má embedovať."
67
  encoded_input = tokenizer(text, return_tensors='tf')
68
  output = model(encoded_input)
 
61
  and in TensorFlow:
62
  ```python
63
  from transformers import RobertaTokenizer, TFRobertaModel
64
+ tokenizer = RobertaTokenizer.from_pretrained('gerulata/slovakbert')
65
+ model = TFRobertaModel.from_pretrained('gerulata/slovakbert')
66
  text = "Text ktorý sa má embedovať."
67
  encoded_input = tokenizer(text, return_tensors='tf')
68
  output = model(encoded_input)