pchopalli commited on
Commit
96bfc89
1 Parent(s): 2fdd6be

Create test.py

Browse files
Files changed (1) hide show
  1. test.py +8 -0
test.py ADDED
@@ -0,0 +1,8 @@
 
 
 
 
 
 
 
 
 
1
+
2
+ !pip install datasets evaluate transformers[sentencepiece]
3
+ from transformers import pipeline
4
+ unmasker = pipeline("fill-mask", model="distilroberta-base")
5
+ unmasker(
6
+ "Unites States is the <mask>",top_k=3,
7
+
8
+ )