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
|