Update app.py

app.py

@@ -30,18 +30,16 @@ def predict(title, summary, tokenizer, model):
     prediction_probs = []
     while sum_probs < 0.95:
         prediction.append(label_to_theme[classes[ind]])
-        prediction_probs.append(probs[classes[ind]])
+        prediction_probs.append(str("{:.2f}".format(100 * 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):
-    frame =  pd.DataFrame({'Topic': prediction, 'Confidence': prediction_probs}).style.hide_index()
-    return frame
+    frame =  pd.DataFrame({'Topic': prediction, 'Confidence': prediction_probs})
+    return frame.to_string(index=False)
     
 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'}