File size: 220 Bytes
96bfc89
 
 
 
 
 
 
 
1
2
3
4
5
6
7
8

!pip install datasets evaluate transformers[sentencepiece]
from transformers import pipeline
unmasker = pipeline("fill-mask", model="distilroberta-base")
unmasker(
          "Unites States is the <mask>",top_k=3,
    
)