Spaces:
Runtime error
Runtime error
working on xlm-roberta indices
Browse files
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"
|