keshan commited on
Commit
07773c6
1 Parent(s): f7305a0

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +2 -2
README.md CHANGED
@@ -27,8 +27,8 @@ You can use this model directly with a pipeline for masked language modeling:
27
  ```py
28
  from transformers import AutoTokenizer, AutoModelWithLMHead, pipeline
29
 
30
- model = BertForMaskedLM.from_pretrained("keshan/SinhalaBERTo")
31
- tokenizer = BertTokenizer.from_pretrained("keshan/SinhalaBERTo")
32
 
33
  fill_mask = pipeline('fill-mask', model=model, tokenizer=tokenizer)
34
 
 
27
  ```py
28
  from transformers import AutoTokenizer, AutoModelWithLMHead, pipeline
29
 
30
+ model = AutoModelWithLMHead.from_pretrained("keshan/SinhalaBERTo")
31
+ tokenizer = AutoTokenizer.from_pretrained("keshan/SinhalaBERTo")
32
 
33
  fill_mask = pipeline('fill-mask', model=model, tokenizer=tokenizer)
34