Update README.md

README.md

@@ -107,7 +107,7 @@ model = Wav2Vec2ForCTC.from_pretrained("facebook/wav2vec2-base-960h").to("cuda")
 processor = Wav2Vec2Processor.from_pretrained("facebook/wav2vec2-base-960h")
 
 def map_to_pred(batch):
-    input_values = processor(batch["audio"]["array"], return_tensors="pt", padding="longest").input_values
+    input_values = processor(batch["audio"][0]["array"], return_tensors="pt", padding="longest").input_values
     with torch.no_grad():
         logits = model(input_values.to("cuda")).logits