tuna2134/sbv2-v2
1
0
Fork 0
Code Issues Pull Requests 1 Actions Packages Projects Releases Wiki Activity

Feat: multi-speaker in app.py

Browse Source
This commit is contained in:
litagin02
2024-01-09 17:42:31 +09:00
parent dd94d23d32
commit 630a873098
2 changed files with 23 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
common/tts_model.py
View File

@@ -199,7 +199,7 @@ class ModelHolder:
def load_model_gr(
self, model_name: str, model_path: str
) -> tuple[gr.Dropdown, gr.Button]:
) -> tuple[gr.Dropdown, gr.Button, gr.Dropdown]:
if model_name not in self.model_files_dict:
raise ValueError(f"Model `{model_name}` is not found")
if model_path not in self.model_files_dict[model_name]:
@@ -210,10 +210,12 @@ class ModelHolder:
style_vec_path=os.path.join(self.root_dir, model_name, "style_vectors.npy"),
device=self.device,
)
speakers = list(self.current_model.spk2id.keys())
styles = list(self.current_model.style2id.keys())
return (
gr.Dropdown(choices=styles, value=styles[0]),
gr.Button(interactive=True, value="音声合成"),
gr.Dropdown(choices=speakers, value=speakers[0]),
)
def update_model_files_gr(self, model_name: str) -> gr.Dropdown: