amansolanki commited on
Commit
023b1b0
·
1 Parent(s): 73dac8c

new demo data

Browse files
Files changed (1) hide show
  1. app.py +7 -3
app.py CHANGED
@@ -57,8 +57,12 @@ st.write(np.round(np.array(ensemble_prediction), 4))
57
  true_label = list(dict(y_true).values())[0]
58
  predicted_label = y_pred_ensemble[0]
59
 
60
- st.write('True Label', true_label)
61
- st.write('Predicted Label', predicted_label)
62
- st.write('0: non-hateful, 1: hateful')
 
 
 
 
63
 
64
  st.button('Random Meme')
 
57
  true_label = list(dict(y_true).values())[0]
58
  predicted_label = y_pred_ensemble[0]
59
 
60
+ if true_label == 0:
61
+ st.write('True Label: non-hateful')
62
+ st.write('True Label: hateful')
63
+
64
+ if predicted_label == 0:
65
+ st.write('Predicted Label: non-hateful)
66
+ st.write('Predicted Label: hateful)
67
 
68
  st.button('Random Meme')