From d80d04636547c269ee98c1505e27bf5e1b434c52 Mon Sep 17 00:00:00 2001
From: dingbig <dingbig@gmail.com>
Date: 星期二, 21 三月 2023 17:11:49 +0800
Subject: [PATCH] fixed token_int is zero bug and add more puncs to sentence
---
funasr/bin/asr_inference_paraformer_vad_punc.py | 2 ++
1 files changed, 2 insertions(+), 0 deletions(-)
diff --git a/funasr/bin/asr_inference_paraformer_vad_punc.py b/funasr/bin/asr_inference_paraformer_vad_punc.py
index 3f57751..ab3e1e3 100644
--- a/funasr/bin/asr_inference_paraformer_vad_punc.py
+++ b/funasr/bin/asr_inference_paraformer_vad_punc.py
@@ -292,6 +292,8 @@
# remove blank symbol id, which is assumed to be 0
token_int = list(filter(lambda x: x != 0 and x != 2, token_int))
+ if len(token_int) == 0:
+ continue
# Change integer-ids to tokens
token = self.converter.ids2tokens(token_int)
--
Gitblit v1.9.1