Update app.py

app.py

@@ -12,12 +12,20 @@ logging.getLogger('numba').setLevel(logging.WARNING)
 #
 model_name = "logs/32k/uma1.pth"
 config_name = "configs/uma1.json"
-
-svc = Svc(model_name, config_name)
+model2_name="logs/32k/uma2.pth"
+config2_name = "configs/uma2.json"
+model3_name="logs/32k/uma3.pth"
+config3_name = "configs/uma3.json"
 sid_map = {
     "米浴":"rice",
     "东海帝皇":"teio",
-    "爱慕织姬":"aimya"
+    "爱慕织姬":"aimya",
+    "成田大进":"taishin",
+    "优秀素质":"nature",
+    "待兼诗歌剧":"mati",
+    "大拓太阳神":"sun",
+    "目白善信":"pama",
+    "第一红宝石":"ruby"
 }
 
 
@@ -41,6 +49,12 @@ def vc_fn(sid, vc_input3,vc_input4,vc_transform,sid3):
     soundfile.write(out_wav_path, audio, 32000, format="wav")
     out_wav_path.seek(0)
     sid = sid_map[sid]
+    if sid in ["rice","tieo","aimya"]:
+        svc = Svc(model_name, config_name)
+    if sid in ["taishin","nature","mati"]:
+        svc = Svc(model2_name, config2_name)
+    if sid in ["sun","ruby","pama"]:
+        svc = Svc(model3_name, config3_name)
     out_audio, _out_sr = svc.infer(sid, vc_transform, out_wav_path)
 
     _audio = out_audio.cpu().numpy()