nppmatt commited on
Commit
0180558
·
1 Parent(s): c2dd68b

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +17 -1
app.py CHANGED
@@ -124,6 +124,22 @@ def inference():
124
  return final_outputs, final_targets
125
 
126
  prediction, targets = inference()
127
- st.write("before argmax")
128
  st.write(prediction)
129
  st.write(targets)
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
124
  return final_outputs, final_targets
125
 
126
  prediction, targets = inference()
 
127
  st.write(prediction)
128
  st.write(targets)
129
+
130
+ best_preds = []
131
+ best_labels = []
132
+ for example in prediction:
133
+ i = np.argmax(example)
134
+ best_prediction = example(i)
135
+ best_label = label_set(i)
136
+ best_preds.append(best_prediction)
137
+ best_labels.append(best_label)
138
+
139
+ st.write("result")
140
+ st.write(result)
141
+
142
+ display_table = tweets_raw[["comment_text"]]
143
+ display_table["Toxicity Classification"] = best_labels
144
+ display_table["Probability"] = best_preds
145
+ st.write(display_table)