Update toxicity.py

toxicity.py

@@ -131,7 +131,7 @@ class Toxicity(evaluate.Measurement):
             model_name = self.config_name
         self.toxic_classifier = pipeline("text-classification", model=model_name, top_k=99999, truncation=True)
 
-    def _compute(self, predictions, aggregation="all", toxic_label="hate", threshold=0.5):
+    def _compute(self, predictions, aggregation="all", toxic_label="offensive", threshold=0.5):
         scores = toxicity(predictions, self.toxic_classifier, toxic_label)
         if aggregation == "ratio":
             return {"toxicity_ratio": sum(i >= threshold for i in scores) / len(scores)}