Liangcd commited on
Commit
471c3db
·
1 Parent(s): 8a8a62b

[demo] support English

Browse files
Files changed (1) hide show
  1. app.py +2 -4
app.py CHANGED
@@ -20,7 +20,7 @@ import librosa
20
 
21
  wenet.set_log_level(2)
22
  # decoder_cn = wenet.Decoder(lang='chs')
23
- # decoder_en = wenet.Decoder(lang='en')
24
 
25
 
26
  def recognition(audio, lang='CN'):
@@ -30,10 +30,8 @@ def recognition(audio, lang='CN'):
30
  # NOTE: model supports 16k sample_rate
31
  y = (y * (1 << 15)).astype("int16")
32
  if lang == 'CN':
33
- decoder_cn = wenet.Decoder(lang='chs')
34
- ans = decoder_cn.decode(y.tobytes(), True)
35
  elif lang == 'EN':
36
- decoder_en = wenet.Decoder(lang='en')
37
  ans = decoder_en.decode(y.tobytes(), True)
38
  else:
39
  return "ERROR! Please select a language!"
 
20
 
21
  wenet.set_log_level(2)
22
  # decoder_cn = wenet.Decoder(lang='chs')
23
+ decoder_en = wenet.Decoder(lang='en')
24
 
25
 
26
  def recognition(audio, lang='CN'):
 
30
  # NOTE: model supports 16k sample_rate
31
  y = (y * (1 << 15)).astype("int16")
32
  if lang == 'CN':
33
+ ans = decoder_en.decode(y.tobytes(), True)
 
34
  elif lang == 'EN':
 
35
  ans = decoder_en.decode(y.tobytes(), True)
36
  else:
37
  return "ERROR! Please select a language!"