Elron Bandel
commited on
Commit
•
8619f75
1
Parent(s):
400e02d
Update waiting message
Browse files
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
|
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]
|