from funasr import AutoModel

model_dir = "/Users/moyoyo/code/Translator/moyoyo_asr_models"
asr_model_path = model_dir + '/speech_seaco_paraformer_large_asr_nat-zh-cn-16k-common-vocab8404-pytorch'
vad_model_path = model_dir + '/speech_fsmn_vad_zh-cn-16k-common-pytorch'
punc_model_path = model_dir + '/punc_ct-transformer_cn-en-common-vocab471067-large'

model = AutoModel(model=asr_model_path)
model_dir = model.export(type="onnx", quantize=True, disable_update=True)
print(model_dir)

model = AutoModel(model=vad_model_path)
model_dir = model.export(type="onnx", quantize=True, disable_update=True)
print(model_dir)

model = AutoModel(model=punc_model_path)
model_dir = model.export(type="onnx", quantize=True, disable_update=True)
print(model_dir)