Spaces:
Sleeping
Sleeping
Update app.py
Browse files
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'])
|