diff --git a/README.md b/README.md index 0c4dbf7..8390a82 100644 --- a/README.md +++ b/README.md @@ -58,7 +58,7 @@ To transcribe audio files the model can be used as a standalone acoustic model a ds = load_dataset("patrickvonplaten/librispeech_asr_dummy", "clean", split="validation") # tokenize - input_values = processor(batch["audio"]["array"], return_tensors="pt", padding="longest").input_values + input_values = processor(ds[0]["audio"]["array"], return_tensors="pt", padding="longest").input_values # retrieve logits logits = model(input_values).logits