From ff83116369ba6afe56430cb0b9d977fbe7f4bced Mon Sep 17 00:00:00 2001
From: Atomie CHEN <atomic_cwh@163.com>
Date: 星期一, 25 三月 2024 09:37:26 +0800
Subject: [PATCH] fix: resolve IndexError when using spk model and the audio contains only 1 segment (#1535)
---
runtime/websocket/bin/funasr-wss-client-2pass.cpp | 5 ++++-
1 files changed, 4 insertions(+), 1 deletions(-)
diff --git a/runtime/websocket/bin/funasr-wss-client-2pass.cpp b/runtime/websocket/bin/funasr-wss-client-2pass.cpp
index 6533dd5..0cbd10e 100644
--- a/runtime/websocket/bin/funasr-wss-client-2pass.cpp
+++ b/runtime/websocket/bin/funasr-wss-client-2pass.cpp
@@ -192,7 +192,10 @@
funasr::Audio audio(1);
int32_t sampling_rate = audio_fs;
std::string wav_format = "pcm";
- if (funasr::IsTargetFile(wav_path.c_str(), "pcm")) {
+ if (funasr::IsTargetFile(wav_path.c_str(), "wav")) {
+ if (!audio.LoadWav(wav_path.c_str(), &sampling_rate, false))
+ return;
+ } else if (funasr::IsTargetFile(wav_path.c_str(), "pcm")) {
if (!audio.LoadPcmwav(wav_path.c_str(), &sampling_rate, false)) return;
} else {
wav_format = "others";
--
Gitblit v1.9.1