qc7 commited on
Commit
ad35274
1 Parent(s): 28d5381

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -1
app.py CHANGED
@@ -43,4 +43,5 @@ button = st.button('Run classifier')
43
 
44
  if button:
45
  probs = forward_pass(title, abstract, tokenizer, model)
46
- st.write(probs)
 
 
43
 
44
  if button:
45
  probs = forward_pass(title, abstract, tokenizer, model)
46
+ micro_df = pd.DataFrame({'Categories': CATEGORIES, 'Cat. Probability': probs})
47
+ st.write(micro_df)