facebook
/

wav2vec2-large-960h-lv60-self

Automatic Speech Recognition

hf-asr-leaderboard

Inference Endpoints

Model card Files Files and versions Community

andreagasparini commited on Jul 25, 2022

Commit

25ca1d9

•

1 Parent(s): d02ba4f

Fixes typo

Files changed (1) hide show

README.md +1 -1

README.md CHANGED Viewed

@@ -103,7 +103,7 @@ processor = Wav2Vec2Processor.from_pretrained("facebook/wav2vec2-large-960h-lv60
 def map_to_pred(batch):
     inputs = processor(batch["audio"]["array"], return_tensors="pt", padding="longest")
-    input_values = inputs.input_values.to s("cuda")
     attention_mask = inputs.attention_mask.to("cuda")
     with torch.no_grad():

 def map_to_pred(batch):
     inputs = processor(batch["audio"]["array"], return_tensors="pt", padding="longest")
+    input_values = inputs.input_values.to("cuda")
     attention_mask = inputs.attention_mask.to("cuda")
     with torch.no_grad():