adorkin commited on
Commit
9548536
·
verified ·
1 Parent(s): dbfe7d9

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +3 -2
README.md CHANGED
@@ -15,9 +15,10 @@ widget:
15
  The EstBERT model is a pretrained BERT<sub>Base</sub> model exclusively trained on Estonian cased corpus on both 128 and 512 sequence length of data.
16
 
17
  ### How to use?
18
- You can use the model transformer library both in tensorflow and pytorch version.
19
- ```
20
  from transformers import AutoTokenizer, AutoModelForMaskedLM
 
21
  tokenizer = AutoTokenizer.from_pretrained("tartuNLP/EstBERT")
22
  model = AutoModelForMaskedLM.from_pretrained("tartuNLP/EstBERT")
23
  ```
 
15
  The EstBERT model is a pretrained BERT<sub>Base</sub> model exclusively trained on Estonian cased corpus on both 128 and 512 sequence length of data.
16
 
17
  ### How to use?
18
+ You can use the model with the transformers library the following way.
19
+ ```python
20
  from transformers import AutoTokenizer, AutoModelForMaskedLM
21
+
22
  tokenizer = AutoTokenizer.from_pretrained("tartuNLP/EstBERT")
23
  model = AutoModelForMaskedLM.from_pretrained("tartuNLP/EstBERT")
24
  ```