Spaces:
Runtime error
Runtime error
JulianHame
commited on
Commit
·
edc1163
1
Parent(s):
2f2fe28
Update app.py
Browse files
app.py
CHANGED
@@ -19,9 +19,9 @@ pipe = pipeline('sentiment-analysis')
|
|
19 |
if selection == "DistilBERT":
|
20 |
pipe = pipeline(model = "distilbert-base-uncased-finetuned-sst-2-english")
|
21 |
if selection == "Toxicity-Classifier":
|
22 |
-
pipe = pipeline(model = "
|
23 |
if selection == "Twitter-roBERTa":
|
24 |
-
pipe = pipeline(model = "
|
25 |
if selection == "SiEBERT":
|
26 |
pipe = pipeline(model = "siebert/sentiment-roberta-large-english")
|
27 |
|
|
|
19 |
if selection == "DistilBERT":
|
20 |
pipe = pipeline(model = "distilbert-base-uncased-finetuned-sst-2-english")
|
21 |
if selection == "Toxicity-Classifier":
|
22 |
+
pipe = pipeline(model = "JulianHame/Toxicity-Classifier")
|
23 |
if selection == "Twitter-roBERTa":
|
24 |
+
pipe = pipeline(model = "unitary/toxic-bert")
|
25 |
if selection == "SiEBERT":
|
26 |
pipe = pipeline(model = "siebert/sentiment-roberta-large-english")
|
27 |
|