iamtarun commited on
Commit
333b712
1 Parent(s): 9dcf373

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +1 -1
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