[demo] Update app.py and add examples

app.py

@@ -1,3 +1,17 @@
+# Copyright (c) 2022 Horizon Robotics. (authors: Binbin Zhang)
+#               2022 Chengdong Liang ([email protected])
+#
+# Licensed under the Apache License, Version 2.0 (the "License");
+# you may not use this file except in compliance with the License.
+# You may obtain a copy of the License at
+#
+#   http://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing, software
+# distributed under the License is distributed on an "AS IS" BASIS,
+# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+# See the License for the specific language governing permissions and
+# limitations under the License.
 
 import json
 import gradio as gr
@@ -5,48 +19,47 @@ import numpy as np
 import wenetruntime as wenet
 
 wenet.set_log_level(2)
-decoder = wenet.Decoder(lang='chs')
+decoder_cn = wenet.Decoder(lang='chs')
 
 
 def recognition(audio):
-    print(audio)
     if audio is None:
         return "Input Error! Please enter one audio!"
     sr, y = audio
     assert sr in [48000, 16000]
     if sr == 48000:  # Optional resample to 16000
         y = (y / max(np.max(y), 1) * 32767)[::3].astype("int16")
-    ans = decoder.decode(y.tobytes(), True)
-    if ans == None:
+    ans = decoder_cn.decode(y.tobytes(), True)
+    if ans is None:
         return "ERROR! No text output! Please try again!"
-    # ans (json)
+    # NOTE: ans (json)
     # {
     #    'nbest' : [{"sentence" : ""}], 'type' : 'final_result
     # }
     ans = json.loads(ans)
-    print(ans)
     txt = ans['nbest'][0]['sentence']
     return txt
 
 
 # input
-inputs = [
-    gr.inputs.Audio(source="microphone",
-                    type="numpy",
-                    label='Speaker#1')
-]
+inputs = gr.inputs.Audio(source="microphone", type="numpy", label='Input audio')
 
 output = gr.outputs.Textbox(label="Output Text")
 
-# examples = ['examples/BAC009S0764W0121.wav']
+examples = [
+   ['examples/BAC009S0767W0127.wav'],
+   ['examples/BAC009S0767W0424.wav'],
+   ['examples/BAC009S0767W0488.wav'],
+]
+
 text = "Speech Recognition in WeNet | 基于 WeNet 的语音识别"
 
 # description
-description = ("WeSpeaker Demo ! Try it with your own voice !")
+description = ("Wenet Demo ! This is a Mandarin streaming speech recognition !")
 
 article = (
     "<p style='text-align: center'>"
-    "<a href='https://github.com/wenet-e2e/wespeaker' target='_blank'>Github: Learn more about WeSpeaker</a>"
+    "<a href='https://github.com/wenet-e2e/wenet' target='_blank'>Github: Learn more about WeNet</a>"
     "</p>")
 
 interface = gr.Interface(
@@ -56,6 +69,7 @@ interface = gr.Interface(
     title=text,
     description=description,
     article=article,
+    examples=examples,
     theme='huggingface',
 )