Harveenchadha
commited on
Commit
•
a1f28e4
1
Parent(s):
1e9cc4e
Update app.py
Browse files
app.py
CHANGED
@@ -27,7 +27,7 @@ def parse_transcription_with_lm(wav_file):
|
|
27 |
input_values = read_file_and_process(wav_file)
|
28 |
|
29 |
with torch.no_grad():
|
30 |
-
logits = model(**input_values).logits.cpu().numpy()
|
31 |
print(logits)
|
32 |
int_result = processor_with_LM.decode(logits = logits, output_word_offsets=False)
|
33 |
print(int_result)
|
|
|
27 |
input_values = read_file_and_process(wav_file)
|
28 |
|
29 |
with torch.no_grad():
|
30 |
+
logits = model(**input_values).logits[0].cpu().numpy()
|
31 |
print(logits)
|
32 |
int_result = processor_with_LM.decode(logits = logits, output_word_offsets=False)
|
33 |
print(int_result)
|