robzchhangte commited on
Commit
48fbb5f
1 Parent(s): 5986ebc

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +3 -3
README.md CHANGED
@@ -32,15 +32,15 @@ Then, import and use MizBERT like other pre-trained models in the library:
32
  ```python
33
  from transformers import AutoTokenizer, AutoModelForMaskedLM
34
 
35
- tokenizer = AutoTokenizer.from_pretrained("robzchhangte/mizbert-25")
36
- model = AutoModelForMaskedLM.from_pretrained("robzchhangte/mizbert-25")
37
  ```
38
 
39
  **To Predict Mask Token**
40
  ```python
41
  from transformers import pipeline
42
 
43
- fill_mask = pipeline("fill-mask", model="robzchhangte/mizbert-25")
44
 
45
  sentence = "Miten kan thiltih [MASK] min teh thin" ##Expected token "atangin". In English: A tree is known by its fruit.
46
  predictions = fill_mask(sentence)
 
32
  ```python
33
  from transformers import AutoTokenizer, AutoModelForMaskedLM
34
 
35
+ tokenizer = AutoTokenizer.from_pretrained("robzchhangte/mizbert")
36
+ model = AutoModelForMaskedLM.from_pretrained("robzchhangte/mizbert")
37
  ```
38
 
39
  **To Predict Mask Token**
40
  ```python
41
  from transformers import pipeline
42
 
43
+ fill_mask = pipeline("fill-mask", model="robzchhangte/mizbert")
44
 
45
  sentence = "Miten kan thiltih [MASK] min teh thin" ##Expected token "atangin". In English: A tree is known by its fruit.
46
  predictions = fill_mask(sentence)