cswamy commited on
Commit
390a6aa
·
1 Parent(s): 5ffb37f

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -30,7 +30,7 @@ def predict(text:str):
30
  # Convert tokens to sentences
31
  options_list = []
32
  for token in top_5_tokens:
33
- option = f"{text.replace(tokenizer.mask_token, tokenizer.decode([token]))}"
34
  options_list.append(option)
35
 
36
  return options_list
 
30
  # Convert tokens to sentences
31
  options_list = []
32
  for token in top_5_tokens:
33
+ option = f"<<< {text.replace(tokenizer.mask_token, tokenizer.decode([token]))} >>>"
34
  options_list.append(option)
35
 
36
  return options_list