Spaces:
Runtime error
Runtime error
Update app.py
Browse files
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 |
-
|
|
|
47 |
micro_df = micro_df.sort_values(by='Cat. Probability', ascending=False)
|
48 |
-
|
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)
|