Harveenchadha commited on
Commit
2ce792c
1 Parent(s): 5626bba

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -29,9 +29,9 @@ def parse_transcription_with_lm(wav_file):
29
  with torch.no_grad():
30
  logits = model(**input_values).logits[0].cpu().numpy()
31
  print(logits)
32
- int_result = processor_with_LM.decode(logits = logits, output_word_offsets=False, skip_special_tokens=True)
33
  print(int_result)
34
- transcription = int_result.text
35
  return transcription
36
 
37
 
 
29
  with torch.no_grad():
30
  logits = model(**input_values).logits[0].cpu().numpy()
31
  print(logits)
32
+ int_result = processor_with_LM.decode(logits = logits, output_word_offsets=False)
33
  print(int_result)
34
+ transcription = int_result.text.replace('<s>','')
35
  return transcription
36
 
37