Update README.md

README.md

@@ -128,12 +128,12 @@ def map_to_pred(batch):
     attention_mask = features.attention_mask.to("cuda")
 
     gen_tokens = model.generate(input_features=input_features, attention_mask=attention_mask)
-    batch["transcription"] = processor.batch_decode(gen_tokens, skip_special_tokens=True)
+    batch["transcription"] = processor.batch_decode(gen_tokens, skip_special_tokens=True)[0]
     return batch
 
 result = librispeech_eval.map(map_to_pred, remove_columns=["audio"])
 
-print("WER:", wer(predictions=result["transcription"], references=result["text"]))
+print("WER:", wer.compute(predictions=result["transcription"], references=result["text"]))
 ```
 
 *Result (WER)*: