Update app.py

app.py

@@ -37,10 +37,12 @@ def predict(title, summary, tokenizer, model):
    
 @st.cache(suppress_st_warning=True) 
 def get_results(prediction, prediction_probs):
-    ans = "Topic:\t\tConfidence:\n"
-    for (topic, prob) in zip(prediction, prediction_probs):
-        ans += topic + "\t\t" + str(prob) + "\n"
-    return ans
+    for prob in prediction_probs:
+        prob = str("{:.2f}".format(100 * prob)) + "%"
+    return pd.DataFrame({
+         'Topic': prediction,
+         'Confidence': prediction_probs,
+         })
     
 label_to_theme = {0: 'Computer science', 1: 'Economics', 2: 'Electrical Engineering and Systems Science', 3: 'Math',
                   4: 'Quantitative biology', 5: 'Quantitative Finance', 6: 'Statistics', 7: 'Physics'}
@@ -54,6 +56,7 @@ title = st.text_area(label='Title', height=100)
 summary = st.text_area(label='Summary (optional)', height=250)
 button = st.button('Run')
 
-prediction, prediction_probs = predict(title, summary, tokenizer, model)
-ans = get_results(prediction, prediction_probs)
-st.markdown(ans)
+if button:
+    prediction, prediction_probs = predict(title, summary, tokenizer, model)
+    ans = get_results(prediction, prediction_probs)
+    st.write('Results: ', ans)