|
from transformers import pipeline |
|
unmasker = pipeline('fill-mask', model='bert-base-uncased') |
|
unmasker("Hello I'm a [MASK] model.") |
|
|
|
from transformers import BertTokenizer, BertModel |
|
tokenizer = BertTokenizer.from_pretrained('bert-base-uncased') |
|
model = BertModel.from_pretrained("bert-base-uncased") |
|
text = "Replace me by any text you'd like." |
|
encoded_input = tokenizer(text, return_tensors='pt') |
|
output = model(**encoded_input) |
|
|
|
|
|
from transformers import BertTokenizer, TFBertModel |
|
tokenizer = BertTokenizer.from_pretrained('bert-base-uncased') |
|
model = TFBertModel.from_pretrained("bert-base-uncased") |
|
text = "Replace me by any text you'd like." |
|
encoded_input = tokenizer(text, return_tensors='tf') |
|
output = model(encoded_input) |