test.py

!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,
    
)