Customize model for similarity

test_parascore.py

@@ -121,10 +121,10 @@ class test_parascore(evaluate.Metric):
             diversity.append(ss)
         return diversity
 
-    def _compute(self, predictions, references, lang='en'):
+    def _compute(self, predictions, references, model_type='microsoft/deberta-xlarge-mnli', lang='en'):
         """Returns the scores"""
         
-        score = self.score.compute(predictions=predictions, references=references, lang=lang)
+        score = self.score.compute(predictions=predictions, references=references, lang=lang, model_type=model_type)
         bert_score = [round(v, 2) for v in score['f1']]
         diversity = self._diverse(predictions, references, lang)