diff --git a/README.md b/README.md index 223bb47..8057378 100644 --- a/README.md +++ b/README.md @@ -100,7 +100,7 @@ def map_to_pred(batch): batch["transcription"] = transcription return batch -result = librispeech_eval.map(map_to_pred, batched=True, batch_size=16, remove_columns=["speech"]) +result = librispeech_eval.map(map_to_pred, batched=True, batch_size=1, remove_columns=["speech"]) print("WER:", wer(result["text"], result["transcription"])) ```