Dogvane Huang
2024-07-02 dfcc5d47587d3e793cbfec2e9509c0e9a9e1732c
runtime/python/libtorch/demo_contextual_paraformer.py
@@ -7,7 +7,7 @@
model = ContextualParaformer(model_dir, batch_size=1, device_id=device_id)  # gpu
wav_path = "{}/.cache/modelscope/hub/{}/example/asr_example.wav".format(Path.home(), model_dir)
hotwords = "你的热词 魔搭 达摩苑"
hotwords = "你的热词 魔搭"
result = model(wav_path, hotwords)
print(result)