Update app.py

app.py

@@ -128,9 +128,10 @@ preds = []
 for i in range(len(res)):
     preds.append(res[i].tolist())
 
+classes = ['Toxic', 'Severe Toxic', 'Obscene', 'Threat', 'Insult', 'Identity Hate']
 
 if st.button('Classify'):
     for i in range(len(res)):
-        st.write(f"{Y_test.columns[i]} : {preds[i]}\n")
+        st.write(f"{classes[i]} : {round(preds[i], 2)}\n")
     st.success('These are the outputs')