mirror of
https://github.com/modelscope/FunASR
synced 2025-09-15 14:48:36 +08:00
99ecaca869
* add python funasr_api supoort * change little to README.md * add core tools stream * modified a little * fix bug for timeout * support for buffer decode * add ffmpeg decode for buffer
71 lines
2.0 KiB
Python
71 lines
2.0 KiB
Python
"""
|
|
Copyright FunASR (https://github.com/alibaba-damo-academy/FunASR). All Rights
|
|
Reserved. MIT License (https://opensource.org/licenses/MIT)
|
|
|
|
2023-2024 by zhaomingwork@qq.com
|
|
"""
|
|
|
|
from funasr_api import FunasrApi
|
|
import wave
|
|
|
|
def recognizer_example():
|
|
# create an recognizer
|
|
rcg = FunasrApi(
|
|
uri="wss://www.funasr.com:10096/"
|
|
)
|
|
# recognizer by filepath
|
|
text=rcg.rec_file("asr_example.mp3")
|
|
print("recognizer by filepath result=",text)
|
|
|
|
|
|
# recognizer by buffer
|
|
# rec_buf(audio_buf,ffmpeg_decode=False),set ffmpeg_decode=True if audio is not PCM or WAV type
|
|
with open("asr_example.wav", "rb") as f:
|
|
audio_bytes = f.read()
|
|
text=rcg.rec_buf(audio_bytes)
|
|
print("recognizer by buffer result=",text)
|
|
|
|
def recognizer_stream_example():
|
|
|
|
rcg = FunasrApi(
|
|
uri="wss://www.funasr.com:10096/"
|
|
)
|
|
#define call_back function for msg
|
|
def on_msg(msg):
|
|
print("stream msg=",msg)
|
|
stream=rcg.create_stream(msg_callback=on_msg)
|
|
|
|
wav_path = "asr_example.wav"
|
|
|
|
with open(wav_path, "rb") as f:
|
|
audio_bytes = f.read()
|
|
|
|
# use FunasrApi's audio2wav to covert other audio to PCM if needed
|
|
#import os
|
|
#from funasr_tools import FunasrTools
|
|
#file_ext=os.path.splitext(wav_path)[-1].upper()
|
|
#if not file_ext =="PCM" and not file_ext =="WAV":
|
|
# audio_bytes=FunasrTools.audio2wav(audio_bytes)
|
|
|
|
stride = int(60 * 10 / 10 / 1000 * 16000 * 2)
|
|
chunk_num = (len(audio_bytes) - 1) // stride + 1
|
|
|
|
for i in range(chunk_num):
|
|
beg = i * stride
|
|
data = audio_bytes[beg : beg + stride]
|
|
stream.feed_chunk(data)
|
|
final_result=stream.wait_for_end()
|
|
print("asr_example.wav stream_result=",final_result)
|
|
|
|
|
|
if __name__ == "__main__":
|
|
|
|
print("example for Funasr_websocket_recognizer")
|
|
|
|
recognizer_stream_example()
|
|
|
|
recognizer_example()
|
|
|
|
|
|
|