Update app.py

app.py

@@ -233,25 +233,24 @@ if __name__ == '__main__':
     hps = utils.get_hparams_from_file('checkpoints/tmp/config.json')
     dev = torch.device("cuda:0" if torch.cuda.is_available() else "cpu")
     models = []
-    schools = ["Nijigasaki High School","Seisho-Nijigasaki","Seisho Music Academy","Rinmeikan Girls School","Frontier School of Arts","Siegfeld Institute of Music"]
+    schools = ["Nijigasaki High School","Seisho Music Academy","Rinmeikan Girls School","Frontier School of Arts","Siegfeld Institute of Music"]
     lan = ["中文","日文","自动","手动"]
     with open("checkpoints/info.json", "r", encoding="utf-8") as f:
         models_info = json.load(f)
+    net_g = SynthesizerTrn(
+            len(symbols),
+            hps.data.filter_length // 2 + 1,
+            hps.train.segment_size // hps.data.hop_length,
+            n_speakers=hps.data.n_speakers,
+            **hps.model).to(dev)
+    _ = net_g.eval()
+    _ = utils.load_checkpoint("checkpoints/tmp/model.pth" , net_g)
     for i in models_info:
         school = models_info[i]
         speakers = school["speakers"]
-        checkpoint = school["checkpoint"]
         phone_dict = {
             symbol: i for i, symbol in enumerate(symbols)
         }
-        net_g = SynthesizerTrn(
-            len(symbols),
-            hps.data.filter_length // 2 + 1,
-            hps.train.segment_size // hps.data.hop_length,
-            n_speakers=hps.data.n_speakers,
-            **hps.model).to(dev)
-        _ = net_g.eval()
-        _ = utils.load_checkpoint(checkpoint, net_g)
         content = []
         for j in speakers:
             sid = int(speakers[j]['sid'])
@@ -267,7 +266,7 @@ if __name__ == '__main__':
             "# <center>少歌&&虹团vits\n"
             "## <center> Please do not generate content that could infringe upon the rights or cause harm to individuals or organizations.\n"
             "## <center> 请不要生成会对个人以及企划造成侵害的内容\n"
-            "## <center>自带chatgpt支持，长句分割支持，srt字幕生成，可修改音频生成路径至live2d语音路径\n"
+            "## <center>,自带chatgpt支持，长句分割支持，srt字幕生成，可修改音频生成路径至live2d语音路径\n"
             "## <center> 数据集版权归官方所有，严禁商用及恶意使用"
         )
         with gr.Tabs():