facebook
/

wav2vec2-base-960h

Automatic Speech Recognition

hf-asr-leaderboard

Eval Results (legacy)

Model card Files Files and versions

Update README.md

#8

by Saeid - opened Apr 22, 2023

base: refs/heads/main

←

from: refs/pr/8

Discussion Files changed

Files changed (1) hide show

README.md +1 -1

README.md CHANGED Viewed

@@ -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
     with torch.no_grad():
         logits = model(input_values.to("cuda")).logits

 processor = Wav2Vec2Processor.from_pretrained("facebook/wav2vec2-base-960h")
 def map_to_pred(batch):
+    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