From e09d87193b57ad5ef99e0b170e78861b91cd2d2c Mon Sep 17 00:00:00 2001
From: Shi Xian <40013335+R1ckShi@users.noreply.github.com>
Date: 星期一, 22 七月 2024 11:32:39 +0800
Subject: [PATCH] Merge pull request #1928 from liugz18/main
---
funasr/auto/auto_model.py | 8 ++++----
1 files changed, 4 insertions(+), 4 deletions(-)
diff --git a/funasr/auto/auto_model.py b/funasr/auto/auto_model.py
index a82f6ed..75324dc 100644
--- a/funasr/auto/auto_model.py
+++ b/funasr/auto/auto_model.py
@@ -511,8 +511,8 @@
sv_output = postprocess(all_segments, None, labels, spk_embedding.cpu())
if self.spk_mode == "vad_segment": # recover sentence_list
sentence_list = []
- for res, vadsegment in zip(restored_data, vadsegments):
- if "timestamp" not in res:
+ for rest, vadsegment in zip(restored_data, vadsegments):
+ if "timestamp" not in rest:
logging.error(
"Only 'iic/speech_paraformer-large-vad-punc_asr_nat-zh-cn-16k-common-vocab8404-pytorch' \
and 'iic/speech_seaco_paraformer_large_asr_nat-zh-cn-16k-common-vocab8404-pytorch'\
@@ -522,8 +522,8 @@
{
"start": vadsegment[0],
"end": vadsegment[1],
- "sentence": res["text"],
- "timestamp": res["timestamp"],
+ "sentence": rest["text"],
+ "timestamp": rest["timestamp"],
}
)
elif self.spk_mode == "punc_segment":
--
Gitblit v1.9.1