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 |
|