Spaces:

deniandriancode
/

kansei

Runtime error

deniandriancode commited on Oct 28, 2023

Commit

f01351d

•

1 Parent(s): 0ce0c70

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -6,7 +6,7 @@ top_k = 10
 masker = pipeline("fill-mask", model="distilroberta-base")
 def get_mask(txt):
- txt = txt.replace("...", masker.tokenizer.mask_token, count=1)
  result = masker(txt, top_k=top_k)
  random.shuffle(result)
  result_seq = []

 masker = pipeline("fill-mask", model="distilroberta-base")
 def get_mask(txt):
+ txt = txt.replace("...", masker.tokenizer.mask_token, 1)
  result = masker(txt, top_k=top_k)
  random.shuffle(result)
  result_seq = []