mohalisad commited on
Commit
8bf417a
·
verified ·
1 Parent(s): 8503845

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +3 -3
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/PartBert-Base")
27
- model = AutoModelForMaskedLM.from_pretrained("PartAI/PartBert-Base")
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/PartBert-Base")
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