Rik Bose commited on
Commit
592a4fd
·
1 Parent(s): 885d77e

clean up outputs a little

Browse files
Files changed (1) hide show
  1. app.py +3 -2
app.py CHANGED
@@ -16,6 +16,7 @@ model = st.selectbox("choose a model", ["roberta-base", "bert-base-uncased", "gp
16
  if text:
17
  data_load_state = st.text('Loading data...')
18
  nlp = get_model(model)
19
- result = nlp(text.replace("*",nlp.tokenizer.mask_token))
20
  data_load_state.text('Loading data...done!')
21
- st.table(result)
 
 
16
  if text:
17
  data_load_state = st.text('Loading data...')
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)