import gradio as gr

from funasr import AutoModel
model = AutoModel(model="paraformer-zh",  vad_model="fsmn-vad",  punc_model="ct-punc",
                  # spk_model="cam++",
                  )

def asr(audio):
    res = model.generate(input=audio, batch_size_s=300)
    return res[0]['text']

with gr.Blocks() as app:
    gr.Markdown("# <center>🌊💕🎶 语音识别</center>")
    inp = gr.Audio(label="请上传一个音频文件", type="filepath")
    btn = gr.Button("一键开启语音识别", variant="primary")
    out = gr.Textbox(label="文字内容", interactive=True)

    btn.click(asr, [inp], [out])

app.launch(share=False, show_error=True)