Harveenchadha
commited on
Commit
•
a958451
1
Parent(s):
23ba586
Update app.py
Browse files
app.py
CHANGED
@@ -47,7 +47,7 @@ def parse_transcription_with_lm(wav_file):
|
|
47 |
with torch.no_grad():
|
48 |
logits = model(**input_values).logits
|
49 |
|
50 |
-
result =
|
51 |
text = result.text
|
52 |
transcription = text[0]
|
53 |
return transcription
|
|
|
47 |
with torch.no_grad():
|
48 |
logits = model(**input_values).logits
|
49 |
|
50 |
+
result = processor_with_LM.batch_decode(logits.cpu().numpy())
|
51 |
text = result.text
|
52 |
transcription = text[0]
|
53 |
return transcription
|