cjell commited on
Commit
ea92f6d
·
1 Parent(s): fff3a21

switching toxic

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -8,7 +8,7 @@ os.environ["HF_HOME"] = "/tmp"
8
 
9
  spam = pipeline("text-classification", model="mrm8488/bert-tiny-finetuned-sms-spam-detection")
10
 
11
- toxic = pipeline("text-classification", model="cardiffnlp/twitter-roberta-base-toxic")
12
 
13
  sentiment = pipeline("text-classification", model = "nlptown/bert-base-multilingual-uncased-sentiment")
14
 
 
8
 
9
  spam = pipeline("text-classification", model="mrm8488/bert-tiny-finetuned-sms-spam-detection")
10
 
11
+ toxic = pipeline("text-classification", model="s-nlp/roberta_toxicity_classifier")
12
 
13
  sentiment = pipeline("text-classification", model = "nlptown/bert-base-multilingual-uncased-sentiment")
14