Liangcd commited on
Commit
8a8a62b
·
1 Parent(s): 8775cab

[demo] support English

Browse files
Files changed (1) hide show
  1. app.py +4 -2
app.py CHANGED
@@ -19,8 +19,8 @@ import wenetruntime as wenet
19
  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,8 +30,10 @@ 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_cn.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!"
 
19
  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
  # 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!"