robzchhangte
commited on
Commit
•
48fbb5f
1
Parent(s):
5986ebc
Update README.md
Browse files
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
|
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)
|
|
|
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)
|