From 053a6bf78505e427489e341805442db20400117a Mon Sep 17 00:00:00 2001 From: litagin02 Date: Sun, 19 May 2024 18:31:45 +0900 Subject: [PATCH] Fix gradio Path serialize error --- gradio_tabs/inference.py | 4 +++- style_bert_vits2/tts_model.py | 6 ++++-- 2 files changed, 7 insertions(+), 3 deletions(-) diff --git a/gradio_tabs/inference.py b/gradio_tabs/inference.py index ef71928..1049db4 100644 --- a/gradio_tabs/inference.py +++ b/gradio_tabs/inference.py @@ -260,7 +260,9 @@ def create_inference_app(model_holder: TTSModelHolder) -> gr.Blocks: ) return app initial_id = 0 - initial_pth_files = model_holder.model_files_dict[model_names[initial_id]] + initial_pth_files = [ + str(f) for f in model_holder.model_files_dict[model_names[initial_id]] + ] with gr.Blocks(theme=GRADIO_THEME) as app: gr.Markdown(initial_md) diff --git a/style_bert_vits2/tts_model.py b/style_bert_vits2/tts_model.py index 456ecbe..9f66874 100644 --- a/style_bert_vits2/tts_model.py +++ b/style_bert_vits2/tts_model.py @@ -483,7 +483,7 @@ class TTSModelHolder: def update_model_files_for_gradio(self, model_name: str): import gradio as gr - model_files = self.model_files_dict[model_name] + model_files = [str(f) for f in self.model_files_dict[model_name]] return gr.Dropdown(choices=model_files, value=model_files[0]) # type: ignore def update_model_names_for_gradio( @@ -493,7 +493,9 @@ class TTSModelHolder: self.refresh() initial_model_name = self.model_names[0] - initial_model_files = self.model_files_dict[initial_model_name] + initial_model_files = [ + str(f) for f in self.model_files_dict[initial_model_name] + ] return ( gr.Dropdown(choices=self.model_names, value=initial_model_name), # type: ignore gr.Dropdown(choices=initial_model_files, value=initial_model_files[0]), # type: ignore