環境変数で読み込むことにした

.gitignore

@@ -2,3 +2,4 @@ target
 models/*.onnx
 models/*.json
 venv
+.env

Cargo.lock

@@ -357,6 +357,12 @@ dependencies = [
  "crypto-common",
 ]
 
+[[package]]
+name = "dotenvy"
+version = "0.15.7"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "1aaf95b3e5c8f23aa320147307562d361db0ae0d51242340f558153b4eb2439b"
+
 [[package]]
 name = "either"
 version = "1.13.0"
@@ -1614,6 +1620,7 @@ version = "0.1.0"
 dependencies = [
  "anyhow",
  "axum",
+ "dotenvy",
  "sbv2_core",
  "serde",
  "tokio",

sbv2_api/Cargo.toml

@@ -6,6 +6,7 @@ edition = "2021"
 [dependencies]
 anyhow.workspace = true
 axum = "0.7.5"
+dotenvy = "0.15.7"
 sbv2_core = { version = "0.1.0", path = "../sbv2_core" }
 serde = { version = "1.0.210", features = ["derive"] }
 tokio = { version = "1.40.0", features = ["full"] }

sbv2_api/src/main.rs

@@ -7,6 +7,7 @@ use axum::{
 };
 use sbv2_core::tts::TTSModel;
 use serde::Deserialize;
+use std::env;
 use std::sync::Arc;
 use tokio::sync::Mutex;
 
@@ -28,9 +29,9 @@ async fn synthesize(
             tts_model
         } else {
             *tts_model = Some(TTSModel::new(
-                "models/debert.onnx",
+                env::var("BERT_MODEL_PATH")?,
-                "models/model_opt.onnx",
+                env::var("MAIN_MODEL_PATH")?,
-                "models/style_vectors.json",
+                env::var("STYLE_VECTORS_PATH")?,
             )?);
             tts_model.as_ref().unwrap()
         };
@@ -47,6 +48,7 @@ struct AppState {
 
 #[tokio::main]
 async fn main() -> anyhow::Result<()> {
+    dotenvy::dotenv().ok();
     let app = Router::new()
         .route("/", get(|| async { "Hello, World!" }))
         .route("/synthesize", post(synthesize))

test.py

@@ -2,7 +2,7 @@ import requests
 
 
 res = requests.post('http://localhost:3000/synthesize', json={
-    "text": "眠たい"
+    "text": "おはよう"
 })
 res.raise_for_status()
 with open('output.wav', 'wb') as f: