Update README.md
Browse files
README.md
CHANGED
@@ -23,8 +23,8 @@ You can use this model directly for Masked Language Modeling using the provided
|
|
23 |
```Python
|
24 |
from transformers import AutoTokenizer, AutoModelForMaskedLM
|
25 |
|
26 |
-
tokenizer = AutoTokenizer.from_pretrained("PartAI/
|
27 |
-
model = AutoModelForMaskedLM.from_pretrained("PartAI/
|
28 |
|
29 |
# prepare input
|
30 |
text = "شهر برلین در کشور <mask> واقع شده است."
|
@@ -39,7 +39,7 @@ It is also possible to use inference pipelines such as below.
|
|
39 |
```Python
|
40 |
from transformers import pipeline
|
41 |
|
42 |
-
inference_pipeline = pipeline('fill-mask', model="PartAI/
|
43 |
inference_pipeline("شهر برلین در کشور <mask> واقع شده است.")
|
44 |
```
|
45 |
|
|
|
23 |
```Python
|
24 |
from transformers import AutoTokenizer, AutoModelForMaskedLM
|
25 |
|
26 |
+
tokenizer = AutoTokenizer.from_pretrained("PartAI/TookaBERT-Base")
|
27 |
+
model = AutoModelForMaskedLM.from_pretrained("PartAI/TookaBERT-Base")
|
28 |
|
29 |
# prepare input
|
30 |
text = "شهر برلین در کشور <mask> واقع شده است."
|
|
|
39 |
```Python
|
40 |
from transformers import pipeline
|
41 |
|
42 |
+
inference_pipeline = pipeline('fill-mask', model="PartAI/TookaBERT-Base")
|
43 |
inference_pipeline("شهر برلین در کشور <mask> واقع شده است.")
|
44 |
```
|
45 |
|