File size: 161 Bytes
c7e1b60
 
604d7bc
 
 
 
1
2
3
4
5
6
import gradio as gr

from transformers import pipeline

unmasker = pipeline("fill-mask")
unmasker("This course will teach you all about <mask> models.", top_k=2)