diff --git a/app.py b/app.py index 281eb27..f556a23 100644 --- a/app.py +++ b/app.py @@ -5,15 +5,15 @@ import gradio as gr import torch import yaml +from gradio_tabs.dataset import create_dataset_app +from gradio_tabs.inference import create_inference_app +from gradio_tabs.merge import create_merge_app +from gradio_tabs.style_vectors import create_style_vectors_app +from gradio_tabs.train import create_train_app from style_bert_vits2.constants import GRADIO_THEME, VERSION from style_bert_vits2.nlp.japanese import pyopenjtalk_worker from style_bert_vits2.nlp.japanese.user_dict import update_dict from style_bert_vits2.tts_model import TTSModelHolder -from webui.dataset import create_dataset_app -from webui.inference import create_inference_app -from webui.merge import create_merge_app -from webui.style_vectors import create_style_vectors_app -from webui.train import create_train_app # このプロセスからはワーカーを起動して辞書を使いたいので、ここで初期化 diff --git a/colab.ipynb b/colab.ipynb index 3da7427..600b0ec 100644 --- a/colab.ipynb +++ b/colab.ipynb @@ -256,7 +256,7 @@ }, "outputs": [], "source": [ - "from webui.train import preprocess_all\n", + "from gradio_tabs.train import preprocess_all\n", "\n", "preprocess_all(\n", " model_name=model_name,\n", @@ -308,7 +308,7 @@ "\n", "\n", "import yaml\n", - "from webui.train import get_path\n", + "from gradio_tabs.train import get_path\n", "\n", "dataset_path, _, _, _, config_path = get_path(model_name)\n", "\n", diff --git a/webui/dataset.py b/gradio_tabs/dataset.py similarity index 100% rename from webui/dataset.py rename to gradio_tabs/dataset.py diff --git a/webui/inference.py b/gradio_tabs/inference.py similarity index 100% rename from webui/inference.py rename to gradio_tabs/inference.py diff --git a/webui/merge.py b/gradio_tabs/merge.py similarity index 100% rename from webui/merge.py rename to gradio_tabs/merge.py diff --git a/webui/style_vectors.py b/gradio_tabs/style_vectors.py similarity index 100% rename from webui/style_vectors.py rename to gradio_tabs/style_vectors.py diff --git a/webui/train.py b/gradio_tabs/train.py similarity index 100% rename from webui/train.py rename to gradio_tabs/train.py diff --git a/preprocess_all.py b/preprocess_all.py index 2eaf846..1ec7d09 100644 --- a/preprocess_all.py +++ b/preprocess_all.py @@ -1,9 +1,9 @@ import argparse from multiprocessing import cpu_count +from gradio_tabs.train import preprocess_all from style_bert_vits2.nlp.japanese import pyopenjtalk_worker from style_bert_vits2.nlp.japanese.user_dict import update_dict -from webui.train import preprocess_all # このプロセスからはワーカーを起動して辞書を使いたいので、ここで初期化