bond005 commited on
Commit
489c772
1 Parent(s): 11bca8b

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +2 -1
README.md CHANGED
@@ -108,7 +108,8 @@ test_dataset = test_dataset.map(speech_file_to_array_fn)
108
  inputs = processor(test_dataset["speech"], sampling_rate=16_000,
109
  return_tensors="pt", padding=True)
110
  with torch.no_grad():
111
- logits = model(inputs.input_values, attention_mask=inputs.attention_mask).logits
 
112
  predicted_sentences = processor.batch_decode(
113
  logits=logits.numpy(),
114
  num_processes=num_processes
 
108
  inputs = processor(test_dataset["speech"], sampling_rate=16_000,
109
  return_tensors="pt", padding=True)
110
  with torch.no_grad():
111
+ logits = model(inputs.input_values,
112
+ attention_mask=inputs.attention_mask).logits
113
  predicted_sentences = processor.batch_decode(
114
  logits=logits.numpy(),
115
  num_processes=num_processes