Update Space (evaluate main: b39b2669)

requirements.txt

@@ -1,2 +1,2 @@
-git+https://github.com/huggingface/evaluate@5aa3982a9a8c86e506860e381d428a64b0cce73b
+git+https://github.com/huggingface/evaluate@b39b26694046961b9ae97a0458b290fc992df2b1
 jiwer

wer.py

@@ -14,7 +14,7 @@
 """ Word Error Ratio (WER) metric. """
 
 import datasets
-from jiwer import compute_measures
+import jiwer
 
 import evaluate
 
@@ -94,13 +94,25 @@ class WER(evaluate.Metric):
         )
 
     def _compute(self, predictions=None, references=None, concatenate_texts=False):
-        if concatenate_texts:
-            return compute_measures(references, predictions)["wer"]
+        if hasattr(jiwer, "compute_measures"):
+            if concatenate_texts:
+                return jiwer.compute_measures(references, predictions)["wer"]
+            else:
+                incorrect = 0
+                total = 0
+                for prediction, reference in zip(predictions, references):
+                    measures = jiwer.compute_measures(reference, prediction)
+                    incorrect += measures["substitutions"] + measures["deletions"] + measures["insertions"]
+                    total += measures["substitutions"] + measures["deletions"] + measures["hits"]
+                return incorrect / total
         else:
-            incorrect = 0
-            total = 0
-            for prediction, reference in zip(predictions, references):
-                measures = compute_measures(reference, prediction)
-                incorrect += measures["substitutions"] + measures["deletions"] + measures["insertions"]
-                total += measures["substitutions"] + measures["deletions"] + measures["hits"]
-            return incorrect / total
+            if concatenate_texts:
+                return jiwer.process_words(references, predictions).wer
+            else:
+                incorrect = 0
+                total = 0
+                for prediction, reference in zip(predictions, references):
+                    measures = jiwer.process_words(reference, prediction)
+                    incorrect += measures.substitutions + measures.deletions + measures.insertions
+                    total += measures.substitutions + measures.deletions + measures.hits
+                return incorrect / total