Harveenchadha commited on
Commit
e40267d
1 Parent(s): 31c8a67

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
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.decode(logits.cpu().numpy())
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