ulysses115 commited on
Commit
979fd10
1 Parent(s): 1be915c

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +16 -1
app.py CHANGED
@@ -1,3 +1,18 @@
1
  import gradio as gr
 
 
2
 
3
- gr.Interface.load("models/nvidia/stt_zh_citrinet_1024_gamma_0_25").launch()
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
  import gradio as gr
2
+ import nemo.collections.asr as nemo_asr
3
+ asr_model = nemo_asr.models.ASRModel.from_pretrained("nvidia/stt_zh_citrinet_1024_gamma_0_25")
4
 
5
+ def trans(audio_path):
6
+ transcriptions = asr_model.transcribe([audio_path])
7
+ print(transcriptions)
8
+ return transcriptions
9
+
10
+ outputs = [gr.outputs.Textbox(label="stt result")]
11
+
12
+ interface = gr.Interface(fn=trans,
13
+ inputs=gr.Audio(source="upload", label="上传音频(长度小于60秒)", type="filepath"),
14
+ outputs=outputs,
15
+ enable_queue=True)
16
+ interface.launch(debug=True)
17
+
18
+ # gr.Interface.load("models/nvidia/stt_zh_citrinet_1024_gamma_0_25").launch()