Views
No views yet
1from transformers import AutoTokenizer, BertForMaskedLM
2
3model = BertForMaskedLM.from_pretrained("emeraldgoose/bert-base-v1-sports")
4tokenizer = AutoTokenizer.from_pretrained("emeraldgoose/bert-base-v1-sports")
5
6text = "산악 자전거 경기는 상대적으로 새로운 [MASK] 1990년대에 활성화 되었다."
7inputs = tokenizer.encode(text, return_tensors='pt')
8
9model.eval()
10outputs = model(inputs)['logits']
11predict = outputs.argmax(-1)[0]
12print(tokenizer.decode(predict))