Spaces:
Running
Running
Liangcd
commited on
Commit
·
8a8a62b
1
Parent(s):
8775cab
[demo] support English
Browse files
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!"
|