ayoni02 commited on
Commit
01964e8
·
verified ·
1 Parent(s): 1d002bd

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -14,7 +14,7 @@ classifier = pipeline("zero-shot-classification", model="facebook/bart-large-mnl
14
 
15
  def main():
16
  text = st.text_input("Enter the text to classify: ") #input("Enter the text to classify: ")
17
- labels = ["TOXIC", "AGGRESSIVE", "HATE SPEECH", "INSULTIVE", "SPAM", "NSFW", "NEUTRAL"]
18
  if text:
19
  results = classifier(text, labels)
20
  #st.write(results['sequence'])
 
14
 
15
  def main():
16
  text = st.text_input("Enter the text to classify: ") #input("Enter the text to classify: ")
17
+ labels = ["TOXIC", "AGGRESSIVE", "HATE SPEECH", "INSULTIVE", "SPAM", "NSFW", "NEUTRAL", "POSITIVE"]
18
  if text:
19
  results = classifier(text, labels)
20
  #st.write(results['sequence'])