qc7 commited on
Commit
bdc83fa
·
1 Parent(s): 93da556

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -2
app.py CHANGED
@@ -43,7 +43,8 @@ button = st.button('Run classifier')
43
 
44
  if button:
45
  probs = forward_pass(title, abstract, tokenizer, model)
46
- micro_df = pd.DataFrame({'Categories': CATEGORIES, 'Cat. Probability': probs})
 
47
  micro_df = micro_df.sort_values(by='Cat. Probability', ascending=False)
48
- micro_df.style.hide_index()
49
  st.write(micro_df)
 
43
 
44
  if button:
45
  probs = forward_pass(title, abstract, tokenizer, model)
46
+ prob_strings = [(str(prob) + '%') for prob in np.round(probs.numpy() * 100, 3)]
47
+ micro_df = pd.DataFrame({'Categories': CATEGORIES, 'Cat. Probability': prob_strings})
48
  micro_df = micro_df.sort_values(by='Cat. Probability', ascending=False)
49
+ df.set_index('Internal ID')
50
  st.write(micro_df)