Update app.py

app.py

@@ -24,6 +24,19 @@ logger = logging.getLogger(__name__)
 limitation = os.getenv("SYSTEM") == "spaces"
 
 
+def get_net_g(model_path: str, version: str, device: str, hps):
+    # 当前版本模型 net_g
+    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(device)
+    _ = net_g.eval()
+    _ = utils.load_checkpoint(model_path, net_g, None, skip_optimizer=True)
+    return net_g
+
 def get_text(text, hps):
     language_str = "JP"
     norm_text, phone, tone, word2ph = clean_text(text, language_str)
@@ -199,14 +212,7 @@ if __name__ == "__main__":
         config_path = hf_hub_download(repo_id=link, filename="config.json")
         model_path = hf_hub_download(repo_id=link, filename=f"{name}.pth")
         hps = utils.get_hparams_from_file(config_path)
-        net_g_ms = 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)
-        utils.load_checkpoint(model_path, net_g_ms, None, skip_optimizer=True)
-        _ = net_g_ms.eval().to(device)
+        net_g_ms = get_net_g(model_path, "v1", device, hps)
         models.append((name, title, example, list(hps.data.spk2id.keys()), net_g_ms, create_tts_fn(net_g_ms, hps)))
 
     # ✅ Gradio UI แบบพร้อมใช้กับ Spaces