Update app.py

app.py

@@ -124,6 +124,22 @@ def inference():
     return final_outputs, final_targets
 
 prediction, targets = inference()
-st.write("before argmax")
 st.write(prediction)
 st.write(targets)
+
+best_preds = []
+best_labels = []
+for example in prediction:
+    i = np.argmax(example)
+    best_prediction = example(i)
+    best_label = label_set(i)
+    best_preds.append(best_prediction)
+    best_labels.append(best_label)
+
+st.write("result")
+st.write(result)
+
+display_table = tweets_raw[["comment_text"]]
+display_table["Toxicity Classification"] = best_labels
+display_table["Probability"] = best_preds
+st.write(display_table)