Update app.py

app.py

@@ -33,17 +33,15 @@ def predict(title, summary, tokenizer, model):
         prediction_probs.append(probs[classes[ind]])
         sum_probs += probs[classes[ind]]
         ind += 1
+    for prob in prediction_probs:
+        prob = str("{:.2f}".format(100 * prob)) + "%"
     
     return prediction, prediction_probs
    
 @st.cache(suppress_st_warning=True) 
 def get_results(prediction, prediction_probs):
-    for prob in prediction_probs:
-        prob = str("{:.2f}".format(100 * prob)) + "%"
-    return pd.DataFrame({
-         'Topic': prediction,
-         'Confidence': prediction_probs,
-         }).style.hide_index()
+    frame =  pd.DataFrame({'Topic': prediction, 'Confidence': prediction_probs}).style.hide_index()
+    return frame
     
 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'}