Update to app

app.py

@@ -30,12 +30,10 @@ def main():
         # default color
         color = "white"
         # get the color based on label
-        if label == "LABEL_0":
+        if label == "not offensive":
             color = "green"
-            label = "No Offense"
-        elif label == "LABEL_1":
+        elif label == "offensive":
             color = "red"
-            label = "Offensive"
         st.session_state.conversation_history.append(('assistant', f"<div style='background-color: {color}; width: auto; height: 50px;'>Label: {label}; Score: {score:.2f}</div>", time.time()))
     # Display chat history
     for sender, message, timestamp in st.session_state.conversation_history: