From 250cbcdcf2f85ef5679cefd14a2eddcab10b9550 Mon Sep 17 00:00:00 2001 From: litagin02 Date: Thu, 28 Dec 2023 14:22:02 +0900 Subject: [PATCH] Delete hf spaces demo env --- app.py | 25 +++---------------------- 1 file changed, 3 insertions(+), 22 deletions(-) diff --git a/app.py b/app.py index 5f6da4d..09ced91 100644 --- a/app.py +++ b/app.py @@ -14,9 +14,6 @@ from config import config from infer import get_net_g, infer from tools.log import logger -is_hf_spaces = os.getenv("SYSTEM") == "spaces" -limit = 100 - class Model: def __init__(self, model_path, config_path, style_vec_path, device): @@ -222,9 +219,6 @@ def tts_fn( emotion, emotion_weight, ): - if is_hf_spaces and len(text) > limit: - raise Exception(f"文字数が{limit}文字を超えています") - assert model_holder.current_model is not None start_time = datetime.datetime.now() @@ -253,7 +247,7 @@ def tts_fn( initial_text = "こんにちは、初めまして。あなたの名前はなんていうの?" -example_local = [ +examples = [ [initial_text, "JP"], [ """あなたがそんなこと言うなんて、私はとっても嬉しい。 @@ -307,18 +301,6 @@ example_local = [ ["语音合成是人工制造人类语音。用于此目的的计算机系统称为语音合成器,可以通过软件或硬件产品实现。", "ZH"], ] -example_hf_spaces = [ - [initial_text, "JP"], - ["えっと、私、あなたのことが好きです!もしよければ付き合ってくれませんか?", "JP"], - ["吾輩は猫である。名前はまだ無い。", "JP"], - ["どこで生れたかとんと見当がつかぬ。なんでも薄暗いじめじめした所でニャーニャー泣いていた事だけは記憶している。", "JP"], - ["やったー!テストで満点取れたよ!私とっても嬉しいな!", "JP"], - ["どうして私の意見を無視するの?許せない!ムカつく!あんたなんか死ねばいいのに。", "JP"], - ["あはははっ!この漫画めっちゃ笑える、見てよこれ、ふふふ、あはは。", "JP"], - ["あなたがいなくなって、私は一人になっちゃって、泣いちゃいそうなほど悲しい。", "JP"], - ["深層学習の応用により、感情やアクセントを含む声質の微妙な変化も再現されている。", "JP"], -] - initial_md = """ # Style-Bert-VITS2 音声合成 @@ -389,13 +371,12 @@ if __name__ == "__main__": model_holder = ModelHolder(model_dir, device) languages = ["JP", "EN", "ZH"] - examples = example_hf_spaces if is_hf_spaces else example_local model_names = model_holder.model_names if len(model_names) == 0: logger.error(f"モデルが見つかりませんでした。{model_dir}にモデルを置いてください。") sys.exit(1) - initial_id = 1 if is_hf_spaces else 0 + initial_id = 0 initial_pth_files = model_holder.model_files_dict[model_names[initial_id]] with gr.Blocks(theme="NoCrypt/miku") as app: @@ -416,7 +397,7 @@ if __name__ == "__main__": choices=initial_pth_files, value=initial_pth_files[0], ) - refresh_button = gr.Button("更新", scale=1, visible=not is_hf_spaces) + refresh_button = gr.Button("更新", scale=1, visible=True) load_button = gr.Button("ロード", scale=1, variant="primary") text_input = gr.TextArea(label="テキスト", value=initial_text)