NA VICTOR O

app.py

@@ -10,10 +10,10 @@ classifier = pipeline("zero-shot-classification", model="facebook/bart-large-mnl
 
 def main():
     text = st.text_input("Enter the text to classify: ") #input("Enter the text to classify: ")
-    labels = ["POSITIVE", "NEGATIVE", "NEUTRAL"]
+    labels = ["TOXIC", "AGGRESSIVE", "HATE SPEECH", "INSULTIVE", "SPAM", "NSFW" "NEUTRAL"]
     if text:
         results = classifier(text, labels)
-        st.write(results['sequence'])
+        #st.write(results['sequence'])
         st.write(results['labels'])
         st.write(results['scores'])
     else: