Rik Bose commited on
Commit
a5ff044
·
1 Parent(s): 592a4fd
Files changed (1) hide show
  1. app.py +2 -1
app.py CHANGED
@@ -18,5 +18,6 @@ if text:
18
  nlp = get_model(model)
19
  result = nlp(text.replace("*", nlp.tokenizer.mask_token))
20
  data_load_state.text('Loading data...done!')
21
- del result["token"]
 
22
  st.table(result)
 
18
  nlp = get_model(model)
19
  result = nlp(text.replace("*", nlp.tokenizer.mask_token))
20
  data_load_state.text('Loading data...done!')
21
+ for c in result:
22
+ del c["token"]
23
  st.table(result)