Spaces:
Runtime error
Runtime error
deniandriancode
commited on
Commit
•
f01351d
1
Parent(s):
0ce0c70
Update app.py
Browse files
app.py
CHANGED
@@ -6,7 +6,7 @@ top_k = 10
|
|
6 |
masker = pipeline("fill-mask", model="distilroberta-base")
|
7 |
|
8 |
def get_mask(txt):
|
9 |
-
txt = txt.replace("...", masker.tokenizer.mask_token,
|
10 |
result = masker(txt, top_k=top_k)
|
11 |
random.shuffle(result)
|
12 |
result_seq = []
|
|
|
6 |
masker = pipeline("fill-mask", model="distilroberta-base")
|
7 |
|
8 |
def get_mask(txt):
|
9 |
+
txt = txt.replace("...", masker.tokenizer.mask_token, 1)
|
10 |
result = masker(txt, top_k=top_k)
|
11 |
random.shuffle(result)
|
12 |
result_seq = []
|