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])
|