sbv2_core 0.2.0-alpha7

Style-Bert-VITSの推論ライブラリ
use std::env;
use std::fs;

#[cfg(feature = "std")]
fn main_inner() -> anyhow::Result<()> {
    use sbv2_core::tts;
    dotenvy::dotenv_override().ok();
    env_logger::init();
    let text = "今日の天気は快晴です。";
    let ident = "aaa";
    let mut tts_holder = tts::TTSModelHolder::new(
        &fs::read(env::var("BERT_MODEL_PATH")?)?,
        &fs::read(env::var("TOKENIZER_PATH")?)?,
        env::var("HOLDER_MAX_LOADED_MODElS")
            .ok()
            .and_then(|x| x.parse().ok()),
    )?;
    let mp = env::var("MODEL_PATH")?;
    let b = fs::read(&mp)?;
    #[cfg(not(feature = "aivmx"))]
    {
        tts_holder.load_sbv2file(ident, b)?;
    }
    #[cfg(feature = "aivmx")]
    {
        if mp.ends_with(".sbv2") {
            tts_holder.load_sbv2file(ident, b)?;
        } else {
            tts_holder.load_aivmx(ident, b)?;
        }
    }

    let audio = tts_holder.easy_synthesize(ident, text, 0, 0, tts::SynthesizeOptions::default())?;
    fs::write("output.wav", audio)?;

    Ok(())
}

#[cfg(not(feature = "std"))]
fn main_inner() -> anyhow::Result<()> {
    Ok(())
}

fn main() {
    if let Err(e) = main_inner() {
        println!("Error: {e}");
    }
}