Update README.md
Browse files
README.md
CHANGED
@@ -19,7 +19,7 @@ model = BertForMaskedLM.from_pretrained('btqkhai/SinoNomBERT')
|
|
19 |
|
20 |
text = '大 [MASK] 百 官 其 𢮿 花 供 饌 皆 用 新 禮'
|
21 |
|
22 |
-
inputs = tokenizer(text, return_tensors="pt"
|
23 |
mask_token_index = torch.where(inputs["input_ids"] == tokenizer.mask_token_id)[1]
|
24 |
# Ground Truth: 宴
|
25 |
logits = model(**inputs).logits
|
|
|
19 |
|
20 |
text = '大 [MASK] 百 官 其 𢮿 花 供 饌 皆 用 新 禮'
|
21 |
|
22 |
+
inputs = tokenizer(text, return_tensors="pt")
|
23 |
mask_token_index = torch.where(inputs["input_ids"] == tokenizer.mask_token_id)[1]
|
24 |
# Ground Truth: 宴
|
25 |
logits = model(**inputs).logits
|