Update README.md

README.md

@@ -40,8 +40,8 @@ input_values = processor(resampled_audio, return_tensors="pt").input_values
 with torch.no_grad():
     logits = model(input_values.to(DEVICE_ID)).logits.cpu()
     
-prediction_ids = torch.argmax(logits, dim=-1)
-output_str = processor.batch_decode(prediction_ids)[0]
+<!-- prediction_ids = torch.argmax(logits, dim=-1) -->
+output_str = processor.batch_decode(logits.numpy())[0]
 print(f"Greedy Decoding: {output_str}")
 ```