From d339765fc2d41ccd9044583df1fcb3fbb1dcb733 Mon Sep 17 00:00:00 2001
From: 游雁 <zhifu.gzf@alibaba-inc.com>
Date: 星期三, 27 十二月 2023 16:48:57 +0800
Subject: [PATCH] funasr1.0
---
funasr/models/bici_paraformer/model.py | 12 +++++-------
1 files changed, 5 insertions(+), 7 deletions(-)
diff --git a/funasr/models/bici_paraformer/model.py b/funasr/models/bici_paraformer/model.py
index 03c8896..c37ba12 100644
--- a/funasr/models/bici_paraformer/model.py
+++ b/funasr/models/bici_paraformer/model.py
@@ -29,7 +29,7 @@
from funasr.utils.timestamp_tools import ts_prediction_lfr6_standard
from funasr.register import tables
from funasr.models.ctc.ctc import CTC
-from funasr.utils.timestamp_tools import time_stamp_sentence
+
from funasr.models.paraformer.model import Paraformer
@@ -321,18 +321,16 @@
text_postprocessed, time_stamp_postprocessed, word_lists = postprocess_utils.sentence_postprocess(
token, timestamp)
- sentences = time_stamp_sentence(None, time_stamp_postprocessed, text_postprocessed)
- result_i = {"key": key[i], "token": token, "text": text, "text_postprocessed": text_postprocessed,
+
+ result_i = {"key": key[i], "text": text_postprocessed,
"timestamp": time_stamp_postprocessed,
- "word_lists": word_lists,
- "sentences": sentences
}
if ibest_writer is not None:
ibest_writer["token"][key[i]] = " ".join(token)
- ibest_writer["text"][key[i]] = text
+ # ibest_writer["text"][key[i]] = text
ibest_writer["timestamp"][key[i]] = time_stamp_postprocessed
- ibest_writer["text_postprocessed"][key[i]] = text_postprocessed
+ ibest_writer["text"][key[i]] = text_postprocessed
else:
result_i = {"key": key[i], "token_int": token_int}
results.append(result_i)
--
Gitblit v1.9.1