Elron Bandel
commited on
Commit
·
400e02d
1
Parent(s):
2df020e
Add predictions title to spinner
Browse files
app.py
CHANGED
|
@@ -126,10 +126,10 @@ if mode == 'Models':
|
|
| 126 |
st.markdown('#### Outputs:')
|
| 127 |
with st.spinner('Running model...'):
|
| 128 |
res = unmasker(input_masked, tokenized=masking_level == 'SubWords', top_k=n_res)
|
| 129 |
-
|
| 130 |
-
|
| 131 |
-
|
| 132 |
-
|
| 133 |
|
| 134 |
|
| 135 |
|
|
|
|
| 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]
|
| 131 |
+
res_table = pd.DataFrame(res)
|
| 132 |
+
st.table(res_table)
|
| 133 |
|
| 134 |
|
| 135 |
|