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

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
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 = "unitary/toxic-bert")
23
  if selection == "Twitter-roBERTa":
24
- pipe = pipeline(model = "cardiffnlp/twitter-roberta-base-sentiment-latest")
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