ArthurZ HF staff rm567 commited on
Commit
18b1956
1 Parent(s): bca8750

missing bracket (#2)

Browse files

- missing bracket (1cc5af76dbb528e27f1038f79cf7adf12b7b1e0c)


Co-authored-by: Robert Maier <rm567@users.noreply.huggingface.co>

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