seba3y commited on
Commit
f4212a0
1 Parent(s): 3a714b7

Upload wav2vec_aligen.py

Browse files
Files changed (1) hide show
  1. wav2vec_aligen.py +1 -1
wav2vec_aligen.py CHANGED
@@ -25,7 +25,7 @@ def load_audio(audio_path, processor):
25
 
26
  @torch.inference_mode()
27
  def get_emissions(input_values, model):
28
- results = model(input_values,).logits
29
  return results
30
 
31
 
 
25
 
26
  @torch.inference_mode()
27
  def get_emissions(input_values, model):
28
+ results = model(input_values,).logits[0]
29
  return results
30
 
31