Update app.py

app.py

@@ -43,7 +43,7 @@ button = st.button('Run classifier')
 
 if button:
     probs = forward_pass(title, abstract, tokenizer, model)
-    prob_strings = [(str(prob) + '%') for prob in np.round(probs.numpy() * 100, 3)]
+    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)
     df.set_index('Internal ID')