Update app.py

app.py

@@ -99,12 +99,11 @@ def main():
             # Convert the list to a numpy array
             arr = np.array(prediction[0])
             max_index = np.argmax(arr)
-
-            st.success("class"+str(max_index))
-            #if prediction > 0.5:
-            #    st.error("Hate Speech Detected")
-            #else:
-            #    st.success("No Hate Speech Detected")
+            if max_index == 1:
+                #negative 
+                st.error("Hate Speech Detected")
+            else
+                st.success("No Hate Speech Detected")
         else:
             st.warning("Please enter some text")