Harveenchadha commited on
Commit
346d452
1 Parent(s): e40267d

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -3
app.py CHANGED
@@ -22,13 +22,13 @@ 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.batch_decode(logits.cpu().numpy())
30
  print(int_result)
31
- transcription = int_result.text
32
  return transcription
33
 
34
 
 
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.decode(logits)
30
  print(int_result)
31
+ transcription = int_result
32
  return transcription
33
 
34