Upload wav2vec_aligen.py

wav2vec_aligen.py

@@ -25,7 +25,7 @@ def load_audio(audio_path, processor):
         
 @torch.inference_mode()
 def get_emissions(input_values, model):
-    results = model(input_values,).logits
+    results = model(input_values,).logits[0]
     return results