ArthurZ HF staff YaYaB commited on
Commit
a356db9
1 Parent(s): 52aafab

Fix typo (forgotten parenthesis) (#2)

Browse files

- Fix typo (forgotten parenthesis) (e08c57728b37f93ccfe8d4ccdccff8133b8efc93)


Co-authored-by: YaYaB <YaYaB@users.noreply.huggingface.co>

Files changed (1) hide show
  1. README.md +1 -1
README.md CHANGED
@@ -115,7 +115,7 @@ The "<|en|>" token is used to specify that the speech is in english and should b
115
  >>> input_features = processor(ds[0]["audio"]["array"], return_tensors="pt").input_features
116
 
117
  >>> # Generate logits
118
- >>> logits = model(input_features, decoder_input_ids = torch.tensor([[50258]]).logits
119
  >>> # take argmax and decode
120
  >>> predicted_ids = torch.argmax(logits, dim=-1)
121
  >>> transcription = processor.batch_decode(predicted_ids)
 
115
  >>> input_features = processor(ds[0]["audio"]["array"], return_tensors="pt").input_features
116
 
117
  >>> # Generate logits
118
+ >>> logits = model(input_features, decoder_input_ids = torch.tensor([[50258]])).logits
119
  >>> # take argmax and decode
120
  >>> predicted_ids = torch.argmax(logits, dim=-1)
121
  >>> transcription = processor.batch_decode(predicted_ids)