keshan
/

SinhalaBERTo

Inference Endpoints

Model card Files Files and versions Community

keshan commited on Jul 11, 2021

Commit

07773c6

•

1 Parent(s): f7305a0

Update README.md

Files changed (1) hide show

README.md +2 -2

README.md CHANGED Viewed

@@ -27,8 +27,8 @@ You can use this model directly with a pipeline for masked language modeling:
 ```py
 from transformers import AutoTokenizer, AutoModelWithLMHead, pipeline
-model = BertForMaskedLM.from_pretrained("keshan/SinhalaBERTo")
-tokenizer = BertTokenizer.from_pretrained("keshan/SinhalaBERTo")
 fill_mask = pipeline('fill-mask', model=model, tokenizer=tokenizer)

 ```py
 from transformers import AutoTokenizer, AutoModelWithLMHead, pipeline
+model = AutoModelWithLMHead.from_pretrained("keshan/SinhalaBERTo")
+tokenizer = AutoTokenizer.from_pretrained("keshan/SinhalaBERTo")
 fill_mask = pipeline('fill-mask', model=model, tokenizer=tokenizer)