Update README.md

README.md

@@ -27,6 +27,7 @@ In a nutshell: This PR adds a new Wav2Vec2WithLMProcessor class as drop-in repla
 The only change from the existing ASR pipeline will be:
 
 ```diff
+import torch
 -from transformers import Wav2Vec2ForCTC, Wav2Vec2Processor
 +from transformers import Wav2Vec2ForCTC, Wav2Vec2ProcessorWithLM
 from datasets import load_dataset
@@ -41,11 +42,12 @@ model = Wav2Vec2ForCTC.from_pretrained("patrickvonplaten/wav2vec2-large-xlsr-53-
 
 input_values = processor(sample["audio"]["array"], return_tensors="pt").input_values
 
-logits = model(input_values).logits
+with torch.no_grad():
+    logits = model(input_values).logits
 
 -prediction_ids = torch.argmax(logits, dim=-1)
 -transcription = processor.batch_decode(prediction_ids)
-+transcription = processor.batch_decode(logits).text
++transcription = processor.batch_decode(logits.cpu().numpy()).text
 
 print(transcription)
 ```