Update README.md
Browse files
README.md
CHANGED
|
@@ -59,7 +59,7 @@ model = HubertForCTC.from_pretrained("facebook/hubert-xlarge-ls960-ft")
|
|
| 59 |
|
| 60 |
ds = load_dataset("patrickvonplaten/librispeech_asr_dummy", "clean", split="validation")
|
| 61 |
|
| 62 |
-
input_values = processor(ds[0]["audio"]["array"]
|
| 63 |
logits = model(input_values).logits
|
| 64 |
predicted_ids = torch.argmax(logits, dim=-1)
|
| 65 |
transcription = processor.decode(predicted_ids[0])
|
|
|
|
| 59 |
|
| 60 |
ds = load_dataset("patrickvonplaten/librispeech_asr_dummy", "clean", split="validation")
|
| 61 |
|
| 62 |
+
input_values = processor(ds[0]["audio"]["array"], return_tensors="pt").input_values # Batch size 1
|
| 63 |
logits = model(input_values).logits
|
| 64 |
predicted_ids = torch.argmax(logits, dim=-1)
|
| 65 |
transcription = processor.decode(predicted_ids[0])
|