File size: 1,687 Bytes
579ca46 | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 | #!/usr/bin/env python3
"""Reference-only helper: SenseVoice ASR on AX650 (libax_asr_api) for QA."""
import ctypes, os, sys
import numpy as np
_BASE = os.environ.get("ASR_BASE", "/root/huyuan/183_workspace/Speech-Translation-ZipVoice")
LIB = os.environ.get("ASR_LIB", _BASE + "/third_party/asr/lib/libax_asr_api.so")
MODEL = os.environ.get("ASR_MODEL", _BASE + "/models/asr_official/sensevoice_ax650")
sys.path.insert(0, os.path.dirname(os.path.abspath(__file__)))
from cutetts_ax import audio as A
def main():
lib = ctypes.CDLL(LIB)
lib.AX_ASR_Init.argtypes = [ctypes.c_int, ctypes.c_char_p]
lib.AX_ASR_Init.restype = ctypes.c_void_p
lib.AX_ASR_RunPCM.argtypes = [ctypes.c_void_p, ctypes.POINTER(ctypes.c_float), ctypes.c_int,
ctypes.c_int, ctypes.c_char_p, ctypes.POINTER(ctypes.c_char_p)]
lib.AX_ASR_RunPCM.restype = ctypes.c_int
lib.AX_ASR_Free.argtypes = [ctypes.c_char_p]
os.environ.setdefault("AX_SENSEVOICE_SHARE_DIR", MODEL)
h = lib.AX_ASR_Init(4, MODEL.encode())
assert h, "init failed"
for path in sys.argv[1:]:
wav, sr = A.read_wav(path)
if sr != 16000:
wav = A.resample_to(wav, sr, 16000)
arr = np.ascontiguousarray(wav, dtype=np.float32)
out = ctypes.c_char_p()
rc = lib.AX_ASR_RunPCM(h, arr.ctypes.data_as(ctypes.POINTER(ctypes.c_float)), arr.size,
16000, b"auto", ctypes.byref(out))
txt = out.value.decode("utf-8", "replace") if out.value else ""
if out.value:
lib.AX_ASR_Free(out)
print(f"{os.path.basename(path)}: rc={rc} {txt.strip()}")
if __name__ == "__main__":
main()
|