ilushado commited on
Commit
0421ede
·
1 Parent(s): 255c656

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -3
app.py CHANGED
@@ -130,11 +130,11 @@ if st.button("Посмотреть топ"):
130
  ans = model(ids.unsqueeze(0), mask.unsqueeze(0), token_type_ids.unsqueeze(0))
131
  elems = [el.item() for el in ans[0].argsort(descending=True)[:10]]
132
  probs = ans[0].softmax(dim=0)
133
- str = ''
134
  for el in elems:
135
- str += str(el) + " : " + str(probs[el]) + "\n"
136
 
137
- st.markdown(str)
138
 
139
 
140
 
 
130
  ans = model(ids.unsqueeze(0), mask.unsqueeze(0), token_type_ids.unsqueeze(0))
131
  elems = [el.item() for el in ans[0].argsort(descending=True)[:10]]
132
  probs = ans[0].softmax(dim=0)
133
+ str_ans = ''
134
  for el in elems:
135
+ str_ans += str(el) + " : " + str(probs[el]) + "\n"
136
 
137
+ st.markdown(str_ans)
138
 
139
 
140