Update app.py

app.py

@@ -131,10 +131,12 @@ if st.button("Посмотреть топ"):
     elems = [el.item() for el in ans[0].argsort(descending=True)[:10]]
     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)
+    st.write(pd.DataFrame({
+    'first column': [idx_to_tag[el] for el in elems],
+    'second column': [round(probs[el].item(), 2) for el in elems],
+    }))
+