kornosk
/

polibertweet-political-twitter-roberta-mlm

masked-token-prediction

Inference Endpoints

Model card Files Files and versions Community

kornosk commited on May 2, 2022

Commit

5a93763

•

1 Parent(s): 7229041

Update README.md

Files changed (1) hide show

README.md +1 -1

README.md CHANGED Viewed

@@ -43,7 +43,7 @@ tokenizer = BertTokenizer.from_pretrained(pretrained_LM_path)
 model = BertForMaskedLM.from_pretrained(pretrained_LM_path)
 # fill mask
-example = "Trump is the <MASK> of USA"
 fill_mask = pipeline('fill-mask', model=model, tokenizer=tokenizer)
 outputs = fill_mask(example)

 model = BertForMaskedLM.from_pretrained(pretrained_LM_path)
 # fill mask
+example = "Trump is the <mask> of USA"
 fill_mask = pipeline('fill-mask', model=model, tokenizer=tokenizer)
 outputs = fill_mask(example)