Refactor: TTSModelInfo changed from TypedDict to Pydantic model

Pydantic models are more robust and properties can be accessed by dots.
This commit is contained in:
tsukumi
2024-03-10 19:21:22 +00:00
parent 859d940916
commit 7f02b0f1d5
3 changed files with 14 additions and 13 deletions

View File

@@ -52,7 +52,7 @@ from style_bert_vits2.nlp.japanese.user_dict import (
rewrite_word,
update_dict,
)
from style_bert_vits2.tts_model import TTSModelHolder
from style_bert_vits2.tts_model import TTSModelHolder, TTSModelInfo
# ---フロントエンド部分に関する処理---
@@ -250,7 +250,7 @@ async def normalize(item: TextRequest):
return normalize_text(item.text)
@router.get("/models_info")
@router.get("/models_info", response_model=list[TTSModelInfo])
def models_info():
return model_holder.models_info