Spaces:
Sleeping
Sleeping
Upload wav2vec_aligen.py
Browse files- 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 |
|