Update app.py

app.py

@@ -242,20 +242,23 @@ 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"]
+    schools = ["Seisho-Nijigasaki","Seisho-betterchinese","Nijigasaki","Nijigasaki-biaobei"]
     lan = ["中文","日文","自动","手动"]
     with open("checkpoints/info.json", "r", encoding="utf-8") as f:
         models_info = json.load(f)
-    checkpoint = models_info["Nijigasaki High School"]["checkpoint"]
-    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)
     for i in models_info:
+        checkpoint = models_info[i]["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)
         school = models_info[i]
         speakers = school["speakers"]
         phone_dict = {