rust/sbv2-api
1
0
Fork 0
mirror of https://github.com/neodyland/sbv2-api.git synced 2026-01-07 06:52:57 +00:00
Code Issues Packages Projects Releases Wiki Activity

use easy synthesize function

Browse Source
This commit is contained in:
tuna2134
2024-09-17 10:44:18 +00:00
parent b9ea462497
commit ccad71c564
1 changed files with 1 additions and 12 deletions
Download Patch File Download Diff File Expand all files Collapse all files

13
sbv2_api/src/main.rs
View File

@@ -49,18 +49,7 @@ async fn synthesize(
log::debug!("processing request: text={text}, ident={ident}, sdp_ratio={sdp_ratio}, length_scale={length_scale}"); log::debug!("processing request: text={text}, ident={ident}, sdp_ratio={sdp_ratio}, length_scale={length_scale}");
let buffer = { let buffer = {
let tts_model = state.tts_model.lock().await; let tts_model = state.tts_model.lock().await;
let (bert_ori, phones, tones, lang_ids) = tts_model.parse_text(&text)?; tts_model.easy_synthesize(&ident, &text, 0, sdp_ratio, length_scale)?
let style_vector = tts_model.get_style_vector(&ident, 0, 1.0)?;
tts_model.synthesize(
ident,
bert_ori.to_owned(),
phones,
tones,
lang_ids,
style_vector,
sdp_ratio,
length_scale,
)?
}; };
Ok(([(CONTENT_TYPE, "audio/wav")], buffer)) Ok(([(CONTENT_TYPE, "audio/wav")], buffer))
} }