Update
This commit is contained in:
@@ -16,6 +16,7 @@ import zipfile
|
||||
from datetime import datetime
|
||||
from io import BytesIO
|
||||
from pathlib import Path
|
||||
import yaml
|
||||
|
||||
import numpy as np
|
||||
import pyopenjtalk
|
||||
@@ -151,6 +152,12 @@ origins = [
|
||||
"http://127.0.0.1:8000",
|
||||
]
|
||||
|
||||
# Get path settings
|
||||
with open(Path("configs/paths.yml"), "r", encoding="utf-8") as f:
|
||||
path_config: dict[str, str] = yaml.safe_load(f.read())
|
||||
# dataset_root = path_config["dataset_root"]
|
||||
assets_root = path_config["assets_root"]
|
||||
|
||||
parser = argparse.ArgumentParser()
|
||||
parser.add_argument("--model_dir", type=str, default="model_assets/")
|
||||
parser.add_argument("--device", type=str, default="cuda")
|
||||
@@ -158,6 +165,9 @@ parser.add_argument("--port", type=int, default=8000)
|
||||
parser.add_argument("--inbrowser", action="store_true")
|
||||
parser.add_argument("--line_length", type=int, default=None)
|
||||
parser.add_argument("--line_count", type=int, default=None)
|
||||
parser.add_argument(
|
||||
"--dir", "-d", type=str, help="Model directory", default=assets_root
|
||||
)
|
||||
|
||||
args = parser.parse_args()
|
||||
device = args.device
|
||||
@@ -296,7 +306,11 @@ def multi_synthesis(request: MultiSynthesisRequest):
|
||||
)
|
||||
audios = []
|
||||
for i, req in enumerate(lines):
|
||||
# Loade model
|
||||
if args.line_length is not None and len(req.text) > args.line_length:
|
||||
raise HTTPException(
|
||||
status_code=400,
|
||||
detail=f"1行の文字数は{args.line_length}文字以下にしてください。",
|
||||
)
|
||||
try:
|
||||
model = model_holder.load_model(
|
||||
model_name=req.model, model_path_str=req.modelFile
|
||||
|
||||
Reference in New Issue
Block a user