Update README.md
Browse files
README.md
CHANGED
@@ -62,7 +62,7 @@ ds = load_dataset("patrickvonplaten/librispeech_asr_dummy", "clean", split="vali
|
|
62 |
ds = ds.map(map_to_array)
|
63 |
|
64 |
# speech data is passed to data processor whose output is then fed to model
|
65 |
-
input_values = processor(ds["speech"][:2], sampling_rate=rate, return_tensors="pt").input_values.to(device)
|
66 |
logits = model(input_values).logits
|
67 |
|
68 |
# retrieve logits
|
|
|
62 |
ds = ds.map(map_to_array)
|
63 |
|
64 |
# speech data is passed to data processor whose output is then fed to model
|
65 |
+
input_values = processor(ds["speech"][:2], sampling_rate=rate, padding="longest", return_tensors="pt").input_values.to(device)
|
66 |
logits = model(input_values).logits
|
67 |
|
68 |
# retrieve logits
|