yijinsheng
2024-11-11 2b747626c898fb6b2ee78038ad052b761a77269d
runtime/python/libtorch/demo_paraformer.py
@@ -1,5 +1,5 @@
from pathlib import Path
from funasr_torch import Paraformer
from funasr_torch.paraformer_bin import Paraformer
model_dir = "iic/speech_paraformer-large_asr_nat-zh-cn-16k-common-vocab8404-pytorch"
model = Paraformer(model_dir, batch_size=1)  # cpu