alvanlii
/

wav2vec2-BERT-cantonese

Automatic Speech Recognition

Inference Endpoints

Model card Files Files and versions Community

alvanlii commited on Apr 3

Commit

1adb7ea

•

1 Parent(s): 3441f07

Update README.md

Files changed (1) hide show

README.md +1 -1

README.md CHANGED Viewed

@@ -61,7 +61,7 @@ with torch.no_grad():
   logits = asr_model(features).logits
 predicted_ids = torch.argmax(logits, dim=-1)
-predictions = processor.batch_decode(predicted_ids)
 ```
 ## Training Hyperparameters

   logits = asr_model(features).logits
 predicted_ids = torch.argmax(logits, dim=-1)
+predictions = processor.batch_decode(predicted_ids, skip_special_tokens=True)
 ```
 ## Training Hyperparameters