Update app.py

app.py

@@ -22,17 +22,18 @@ class VitsGradio:
     def __init__(self):
         self.dev = torch.device("cuda:0" if torch.cuda.is_available() else "cpu")
         self.lan = ["中文","日文","自动","手动"]
-        self.idols = ["高咲侑","歩夢","かすみ","しずく","果林","愛","彼方","せつ菜","璃奈","栞子","エマ","ランジュ","ミア","華恋","まひる","なな","クロディーヌ","ひかり",'純那',"香子","真矢","双葉","ミチル","メイファン","やちよ","晶","いちえ","ゆゆ子","塁","珠緒","あるる","ララフィン","美空","静羽","あるる"]
+        self.idols = ["c1","c2","高咲侑","歩夢","かすみ","しずく","果林","愛","彼方","せつ菜","璃奈","栞子","エマ","ランジュ","ミア","華恋","まひる","なな","クロディーヌ","ひかり",'純那',"香子","真矢","双葉","ミチル","メイファン","やちよ","晶","いちえ","ゆゆ子","塁","珠緒","あるる","ララフィン","美空","静羽","あるる"]
         self.modelPaths = []
-        self.loadCk('tmp')
         for root,dirs,files in os.walk("checkpoints"):
             for dir in dirs:
                 self.modelPaths.append(dir)
         with gr.Blocks() as self.Vits:
             gr.Markdown(
-            "## <center> Lovelive虹团-少女歌剧VITS\n"
+            "## <center> Lovelive虹团中日双语VITS\n"
             "### <center> 请不要生成会对个人以及企划造成侵害的内容\n"
-            "<div align='center'>由于神秘bug，无法切换模型，轻量化版本 https://huggingface.co/spaces/Mahiruoshi/lovelive-ShojoKageki-vits </div>")
+            "<div align='center'>目前有标贝普通话版，去标贝版，少歌模型还是大饼状态</div>"
+            '<div align="center"><a>参数说明:由于爱抖露们过于有感情，合成日语时建议将噪声比例调节至0.2-0.3区间，噪声偏差对应着每个字之间的间隔，对普通话影响较大，duration代表整体语速</div>'
+            '<div align="center"><a>合成前请先选择模型，否则第一次合成不一定成功。长段落/小说合成建议colab或本地运行</div>')
             with gr.Tab("TTS合成"):
                 with gr.Row():
                     with gr.Column():
@@ -43,11 +44,17 @@ class VitsGradio:
                                 input3 = gr.Dropdown(label="Speaker", choices=self.idols, value="歩夢", interactive=True)
                                 btnVC = gr.Button("Submit")
                             with gr.Column():
-                                input4 = gr.Slider(minimum=0, maximum=1.0, label="更改噪声比例(noise scale)，以控制情感", value=0.6)
-                                input5 = gr.Slider(minimum=0, maximum=1.0, label="更改噪声偏差(noise scale w)，以控制音素长短", value=0.668)
+                                input4 = gr.Slider(minimum=0, maximum=1.0, label="更改噪声比例(noise scale)，以控制情感", value=0.267)
+                                input5 = gr.Slider(minimum=0, maximum=1.0, label="更改噪声偏差(noise scale w)，以控制音素长短", value=0.7)
                                 input6 = gr.Slider(minimum=0.1, maximum=10, label="duration", value=1)
                                 output1 = gr.Audio(label="采样率22050")
                 btnVC.click(self.infer, inputs=[input1, input2, input3, input4, input5, input6], outputs=[output1])
+            with gr.Tab("选择模型"):
+                with gr.Column():
+                    modelstrs = gr.Dropdown(label = "模型", choices = self.modelPaths, value = self.modelPaths[0], type = "value")
+                    btnMod = gr.Button("载入模型")
+                    statusa = gr.TextArea()
+                    btnMod.click(self.loadCk, inputs=[modelstrs], outputs = [statusa])
             with gr.Tab("Voice Conversion"):
                 gr.Markdown("""
                                 录制或上传声音，并选择要转换的音色。
@@ -73,8 +80,8 @@ class VitsGradio:
                                 input3 = gr.Dropdown(label="Speaker", choices=self.idols, value="歩夢", interactive=True)
                                 btnVC = gr.Button("Submit")
                             with gr.Column():
-                                input4 = gr.Slider(minimum=0, maximum=1.0, label="更改噪声比例(noise scale)，以控制情感", value=0.6)
-                                input5 = gr.Slider(minimum=0, maximum=1.0, label="更改噪声偏差(noise scale w)，以控制音素长短", value=0.668)
+                                input4 = gr.Slider(minimum=0, maximum=1.0, label="更改噪声比例(noise scale)，以控制情感", value=0.267)
+                                input5 = gr.Slider(minimum=0, maximum=1.0, label="更改噪声偏差(noise scale w)，以控制音素长短", value=0.7)
                                 input6 = gr.Slider(minimum=0.1, maximum=10, label="Duration", value=1)
                                 output1 = gr.Audio(label="采样率22050")
                                 subtitle = gr.outputs.File(label="字幕文件：subtitles.srt")