Harveenchadha
commited on
Commit
•
e40267d
1
Parent(s):
31c8a67
Update app.py
Browse files
app.py
CHANGED
@@ -22,11 +22,11 @@ def read_file(wav):
|
|
22 |
|
23 |
|
24 |
def parse_transcription_with_lm(wav_file):
|
25 |
-
input_values = read_file_and_process(wav_file)
|
26 |
|
27 |
with torch.no_grad():
|
28 |
logits = model(**input_values).logits
|
29 |
-
int_result = processor.
|
30 |
print(int_result)
|
31 |
transcription = int_result.text
|
32 |
return transcription
|
|
|
22 |
|
23 |
|
24 |
def parse_transcription_with_lm(wav_file):
|
25 |
+
input_values = [read_file_and_process(wav_file)]
|
26 |
|
27 |
with torch.no_grad():
|
28 |
logits = model(**input_values).logits
|
29 |
+
int_result = processor.batch_decode(logits.cpu().numpy())
|
30 |
print(int_result)
|
31 |
transcription = int_result.text
|
32 |
return transcription
|