Update app.py

app.py

@@ -44,7 +44,8 @@ button = st.button('Run classifier')
 if button:
     probs = forward_pass(title, abstract, tokenizer, model)
     prob_strings = [(str(prob) + '%') for prob in np.round(probs * 100, 3)]
-    micro_df = pd.DataFrame({'Categories': CATEGORIES, 'Cat. Probability': prob_strings})
-    micro_df = micro_df.sort_values(by='Cat. Probability', ascending=False)
+    micro_df = pd.DataFrame({'Categories': CATEGORIES, 'Cat. Probability': prob_strings, 'sort_probs': probs})
+    micro_df = micro_df.sort_values(by='sort_probs', ascending=False)
+    micro_df = micro_df[['Categories', 'Cat.Probability']]
     micro_df.index.name = 'Internal ID'
     st.write(micro_df)