Spaces:
Runtime error
Runtime error
File size: 2,254 Bytes
79f9f38 |
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 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 |
'''
Reference: https://github.com/alibaba-damo-academy/FunASR
pip install funasr
pip install modelscope
pip install -U rotary_embedding_torch
'''
try:
from funasr import AutoModel
except:
print("如果想使用FunASR，请先安装funasr，若使用Whisper，请忽略此条信息")
import os
import sys
sys.path.append('./')
from src.cost_time import calculate_time
class FunASR:
def __init__(self) -> None:
# 定义模型的自定义路径
model_path = "FunASR/speech_seaco_paraformer_large_asr_nat-zh-cn-16k-common-vocab8404-pytorch"
vad_model_path = "FunASR/speech_fsmn_vad_zh-cn-16k-common-pytorch"
punc_model_path = "FunASR/punc_ct-transformer_zh-cn-common-vocab272727-pytorch"
# 检查文件是否存在于 FunASR 目录下
model_exists = os.path.exists(model_path)
vad_model_exists = os.path.exists(vad_model_path)
punc_model_exists = os.path.exists(punc_model_path)
# Modelscope AutoDownload
self.model = AutoModel(
model=model_path if model_exists else "paraformer-zh",
vad_model=vad_model_path if vad_model_exists else "fsmn-vad",
punc_model=punc_model_path if punc_model_exists else "ct-punc-c",
)
# 自定义路径
# self.model = AutoModel(model="FunASR/speech_seaco_paraformer_large_asr_nat-zh-cn-16k-common-vocab8404-pytorch", # model_revision="v2.0.4",
# vad_model="FunASR/speech_fsmn_vad_zh-cn-16k-common-pytorch", # vad_model_revision="v2.0.4",
# punc_model="FunASR/punc_ct-transformer_zh-cn-common-vocab272727-pytorch", # punc_model_revision="v2.0.4",
# # spk_model="cam++", spk_model_revision="v2.0.2",
# )
@calculate_time
def transcribe(self, audio_file):
res = self.model.generate(input=audio_file,
batch_size_s=300)
print(res)
return res[0]['text']
if __name__ == "__main__":
import os
# 创建ASR对象并进行语音识别
audio_file = "output.wav" # 音频文件路径
if not os.path.exists(audio_file):
os.system('edge-tts --text "hello" --write-media output.wav')
asr = FunASR()
print(asr.transcribe(audio_file)) |