Spaces:

ilushado
/

article_roberta_classifier

Runtime error

ilushado commited on Apr 17, 2023

Commit

1edf728

1 Parent(s): 0421ede

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -132,7 +132,7 @@ if st.button("Посмотреть топ"):
     probs = ans[0].softmax(dim=0)
     str_ans = ''
     for el in elems:
-        str_ans += str(el) + " : " + str(probs[el]) + "\n"
     st.markdown(str_ans)

     probs = ans[0].softmax(dim=0)
     str_ans = ''
     for el in elems:
+        str_ans += idx_to_tag[el] + " : " + str(probs[el].item()) + "\n"
     st.markdown(str_ans)