nppmatt commited on
Commit
1d8cd54
1 Parent(s): bf492a2

working on xlm-roberta indices

Browse files
Files changed (1) hide show
  1. app.py +2 -0
app.py CHANGED
@@ -25,6 +25,8 @@ elif (option == "DistilBERT"):
25
  elif (option == "XLM-RoBERTa"):
26
  tokenizerPath = "unitary/multilingual-toxic-xlm-roberta"
27
  modelPath = "unitary/multilingual-toxic-xlm-roberta"
 
 
28
  else:
29
  tokenizerPath = "s-nlp/roberta_toxicity_classifier"
30
  modelPath = "s-nlp/roberta_toxicity_classifier"
 
25
  elif (option == "XLM-RoBERTa"):
26
  tokenizerPath = "unitary/multilingual-toxic-xlm-roberta"
27
  modelPath = "unitary/multilingual-toxic-xlm-roberta"
28
+ neutralIndex = 0
29
+ toxicIndex = 1
30
  else:
31
  tokenizerPath = "s-nlp/roberta_toxicity_classifier"
32
  modelPath = "s-nlp/roberta_toxicity_classifier"