Update app.py

app.py

@@ -40,11 +40,11 @@ def get_net_g(model_path: str, version: str, device: str, hps):
     _ = utils.load_checkpoint(model_path, net_g, None, skip_optimizer=True)
     return net_g
 
-def get_text(text, hps):
-    language_str = "JP"
+def get_text(text, language_str, hps, device, style_text=None, style_weight=0.7):
+    style_text = None if style_text == "" else style_text
+    # 在此处实现当前版本的get_text
     norm_text, phone, tone, word2ph = clean_text(text, language_str)
     phone, tone, language = cleaned_text_to_sequence(phone, tone, language_str)
-
     if hps.data.add_blank:
         phone = commons.intersperse(phone, 0)
         tone = commons.intersperse(tone, 0)
@@ -52,18 +52,17 @@ def get_text(text, hps):
         for i in range(len(word2ph)):
             word2ph[i] = word2ph[i] * 2
         word2ph[0] += 1
-
-    bert = get_bert(norm_text, word2ph, language_str, device)
+    bert = get_bert(norm_text, word2ph, language_str, device, style_text, style_weight)
     del word2ph
-    assert bert.shape[-1] == len(phone), phone
 
-    ja_bert = bert
-    bert = torch.zeros(1024, len(phone))
+    assert bert.shape[-1] == len(
+        phone
+    ), f"Bert seq len {bert.shape[-1]} != {len(phone)}"
 
     phone = torch.LongTensor(phone)
     tone = torch.LongTensor(tone)
     language = torch.LongTensor(language)
-    return bert, ja_bert, phone, tone, language
+    return bert, phone, tone, language
 
 
 def infer(