From 2ab8b7f473d37299f6774e98179767fe4be6a0f2 Mon Sep 17 00:00:00 2001 From: lzr265946 Date: Tue, 21 Mar 2023 12:55:16 +0800 Subject: [PATCH] modify abbr postprocess --- funasr/utils/postprocess_utils.py | 7 ++++--- 1 file changed, 4 insertions(+), 3 deletions(-) diff --git a/funasr/utils/postprocess_utils.py b/funasr/utils/postprocess_utils.py index 2475548f1b..40756d8572 100644 --- a/funasr/utils/postprocess_utils.py +++ b/funasr/utils/postprocess_utils.py @@ -106,17 +106,18 @@ def abbr_dispose(words: List[Any], time_stamp: List[List] = None) -> List[Any]: if num in abbr_begin: if time_stamp is not None: begin = time_stamp[ts_nums[num]][0] - word_lists.append(words[num].upper()) + abbr_word = words[num].upper() num += 1 while num < words_size: if num in abbr_end: - word_lists.append(words[num].upper()) + abbr_word += words[num].upper() last_num = num break else: if words[num].encode('utf-8').isalpha(): - word_lists.append(words[num].upper()) + abbr_word += words[num].upper() num += 1 + word_lists.append(abbr_word) if time_stamp is not None: end = time_stamp[ts_nums[num]][1] ts_lists.append([begin, end])