Update app.py

app.py

@@ -18,7 +18,7 @@ class VitsGradio:
     def __init__(self):
         self.dev = torch.device("cuda:0" if torch.cuda.is_available() else "cpu")
         self.lan = ["中文","日文","自动","手动"]
-        self.idols = ["歩夢","かすみ","しずく","果林","愛","せつ菜","璃奈","栞子","エマ","ランジュ","ミア","派蒙"]
+        self.idols = ["歩夢","かすみ","しずく","果林","愛","せつ菜","璃奈","栞子","エマ","ランジュ","ミア","派蒙","华恋","真昼","天堂","双叶"]
         self.modelPaths = []
         for root,dirs,files in os.walk("checkpoints"):
             for dir in dirs:
@@ -151,6 +151,18 @@ class VitsGradio:
         elif speaker == "派蒙":
             spk = 16
             return spk
+        elif speaker == "华恋":
+            spk = 21
+            return spk
+        elif speaker == "真昼":
+            spk = 20
+            return spk
+        elif speaker == "天堂":
+            spk = 22
+            return spk
+        elif speaker == "双叶":
+            spk = 23
+            return spk
     
     def sle(self,language,text):
         text = text.replace('\n','。').replace(' ',',')