Update app.py

app.py

@@ -52,5 +52,26 @@ if(identity_hate > highest_class_rating):
 st.write("---")
 st.write("Overall toxicity rating: " +str(toxicity))
 st.write("---")
-st.write("Highest toxicity class:" +str(highest_class))
-st.write("Rating: " +str(highest_class_rating))
+st.write("Classifications:")
+if(toxicity_severe > 0.5)
+    st.write("Severely toxic - " +str(toxicity_severe))
+if(obscene > 0.5)
+    st.write("Obscene - " +str(obscene))
+if(threat > 0.5)
+    st.write("Threat - " +str(threat))
+if(insult > 0.5)
+    st.write("Insult - " +str(insult))
+if(identity_hate > 0.5)
+    st.write("Identity hate - " +str(identity_hate))
+st.write("---")
+st.write("Invalid classifications:")
+if(toxicity_severe <= 0.5)
+    st.write("Severely toxic - " +str(toxicity_severe))
+if(obscene <= 0.5)
+    st.write("Obscene - " +str(obscene))
+if(threat <= 0.5)
+    st.write("Threat - " +str(threat))
+if(insult <= 0.5)
+    st.write("Insult - " +str(insult))
+if(identity_hate <= 0.5)
+    st.write("Identity hate - " +str(identity_hate))