ArthurZ HF staff Aaryan commited on
Commit
b8d5e6b
1 Parent(s): 3eff569

Minor Update to Readme.md (#7)

Browse files

- Minor Update to Readme.md (dfd8d9c05c696f19163c5ccb168bf5a1676a12ed)


Co-authored-by: Aaryan Y V S <Aaryan@users.noreply.huggingface.co>

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