Shi Xian
2024-06-18 6c467e6f0abfc6d20d0621fbbf67b4dbd81776cc
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