Ahmadzei's picture
added 3 more tables for large emb model
5fa1a76
If you like, you can use the top_k parameter to specify how many predictions to return:
from transformers import pipeline
mask_filler = pipeline("fill-mask", "username/my_awesome_eli5_mlm_model")
mask_filler(text, top_k=3)
[{'score': 0.5150994658470154,
'token': 21300,
'token_str': ' spiral',
'sequence': 'The Milky Way is a spiral galaxy.