Nadiveedishravanreddy commited on
Commit
604d7bc
·
1 Parent(s): 873c6c5

Create fill-mask.py

Browse files
Files changed (1) hide show
  1. fill-mask.py +4 -0
fill-mask.py ADDED
@@ -0,0 +1,4 @@
 
 
 
 
 
1
+ from transformers import pipeline
2
+
3
+ unmasker = pipeline("fill-mask")
4
+ unmasker("This course will teach you all about <mask> models.", top_k=2)