JulianHame commited on
Commit
2f2fe28
1 Parent(s): b5f0ee1

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -19,7 +19,7 @@ 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 = "JulianHame/toxicity-classifier")
23
  if selection == "Twitter-roBERTa":
24
  pipe = pipeline(model = "cardiffnlp/twitter-roberta-base-sentiment-latest")
25
  if selection == "SiEBERT":
 
19
  if selection == "DistilBERT":
20
  pipe = pipeline(model = "distilbert-base-uncased-finetuned-sst-2-english")
21
  if selection == "Toxicity-Classifier":
22
+ pipe = pipeline(model = "unitary/toxic-bert")
23
  if selection == "Twitter-roBERTa":
24
  pipe = pipeline(model = "cardiffnlp/twitter-roberta-base-sentiment-latest")
25
  if selection == "SiEBERT":