Spaces:

ilushado
/

article_roberta_classifier

Runtime error

ilushado commited on Apr 17, 2023

Commit

72e9f79

1 Parent(s): 1edf728

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -132,9 +132,9 @@ if st.button("Посмотреть топ"):
     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)

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