[fix] fix error

app.py

@@ -16,24 +16,26 @@
 import json
 import gradio as gr
 import torch
-import wenetruntime as wenet
+import wenet
 import librosa
 
 wenet.set_log_level(2)
-chs_decoder = wenet.Decoder(lang='chs')
-en_decoder = wenet.Decoder(lang='en')
+# TODO: add hotword
+chs_model = wenet.load_model('chinese')
+en_model = wenet.load_model('english')
 
 
 def recognition(audio, lang='CN'):
     if audio is None:
         return "Input Error! Please enter one audio!"
-    y, _ = librosa.load(audio, sr=16000)
+    # y, _ = librosa.load(audio, sr=16000)
     # NOTE: model supports 16k sample_rate
-    y = (y * (1 << 15)).astype("int16")
+    # y = (y * (1 << 15)).astype("int16")
     if lang == 'CN':
-        ans = chs_decoder.decode(y.tobytes(), True)
+        # ans = chs_decoder.decode(y.tobytes(), True)
+        ans = chs_model.transcribe(audio)
     elif lang == 'EN':
-        ans = en_decoder.decode(y.tobytes(), True)
+        ans = en_model.transcribe(audio)
     else:
         return "ERROR! Please select a language!"
 
@@ -43,8 +45,7 @@ def recognition(audio, lang='CN'):
     # {
     #    'nbest' : [{"sentence" : ""}], 'type' : 'final_result
     # }
-    ans = json.loads(ans)
-    txt = ans['nbest'][0]['sentence']
+    txt = ans['text']
     return txt
 
 

requirements.txt

@@ -1,3 +1,3 @@
-wenetruntime
+wenet @ git+https://github.com/wenet-e2e/wenet.git
 gradio
 librosa