From c3867c850182abaab1596dbe20ac8aba2df8f5c6 Mon Sep 17 00:00:00 2001
From: lingyunfly <121302812+lingyunfly@users.noreply.github.com>
Date: 星期二, 14 三月 2023 13:18:36 +0800
Subject: [PATCH] Update infer_online.py
---
funasr/bin/punc_inference_launch.py | 3 +++
1 files changed, 3 insertions(+), 0 deletions(-)
diff --git a/funasr/bin/punc_inference_launch.py b/funasr/bin/punc_inference_launch.py
index 53db1df..e7e3f15 100755
--- a/funasr/bin/punc_inference_launch.py
+++ b/funasr/bin/punc_inference_launch.py
@@ -75,6 +75,9 @@
if mode == "punc":
from funasr.bin.punctuation_infer import inference_modelscope
return inference_modelscope(**kwargs)
+ if mode == "punc_VadRealtime":
+ from funasr.bin.punctuation_infer_vadrealtime import inference_modelscope
+ return inference_modelscope(**kwargs)
else:
logging.info("Unknown decoding mode: {}".format(mode))
return None
--
Gitblit v1.9.1