Elron Bandel commited on
Commit
8619f75
1 Parent(s): 400e02d

Update waiting message

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -124,7 +124,7 @@ if mode == 'Models':
124
  unsafe_allow_html=True,
125
  )
126
  st.markdown('#### Outputs:')
127
- with st.spinner('Running model...'):
128
  res = unmasker(input_masked, tokenized=masking_level == 'SubWords', top_k=n_res)
129
  if res:
130
  res = [{'Prediction':r['token_str'], 'Completed Sentence':r['sequence'].replace('[SEP]', '').replace('[CLS]', ''), 'Score':r['score']} for r in res]
 
124
  unsafe_allow_html=True,
125
  )
126
  st.markdown('#### Outputs:')
127
+ with st.spinner(f'Running {model_tags[0]} (may take a minute)...'):
128
  res = unmasker(input_masked, tokenized=masking_level == 'SubWords', top_k=n_res)
129
  if res:
130
  res = [{'Prediction':r['token_str'], 'Completed Sentence':r['sequence'].replace('[SEP]', '').replace('[CLS]', ''), 'Score':r['score']} for r in res]