Minor Update to Readme.md (#7)
Browse files- Minor Update to Readme.md (dfd8d9c05c696f19163c5ccb168bf5a1676a12ed)
Co-authored-by: Aaryan Y V S <[email protected]>
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)
|