Spaces:
Sleeping
Sleeping
ayoni02
commited on
Commit
·
e324a72
1
Parent(s):
6b74b06
edits
Browse files
app.py
CHANGED
@@ -10,7 +10,7 @@ classifier = pipeline("zero-shot-classification", model="facebook/bart-large-mnl
|
|
10 |
|
11 |
def main():
|
12 |
text = st.text_input("Enter the text to classify: ") #input("Enter the text to classify: ")
|
13 |
-
labels = ["TOXIC", "AGGRESSIVE", "HATE SPEECH", "INSULTIVE", "SPAM", "NSFW" "NEUTRAL"]
|
14 |
if text:
|
15 |
results = classifier(text, labels)
|
16 |
#st.write(results['sequence'])
|
|
|
10 |
|
11 |
def main():
|
12 |
text = st.text_input("Enter the text to classify: ") #input("Enter the text to classify: ")
|
13 |
+
labels = ["TOXIC", "AGGRESSIVE", "HATE SPEECH", "INSULTIVE", "SPAM", "NSFW", "NEUTRAL"]
|
14 |
if text:
|
15 |
results = classifier(text, labels)
|
16 |
#st.write(results['sequence'])
|