Update README.md

README.md

@@ -194,21 +194,21 @@ with open("configs/paths.yml", "w", encoding="utf-8") as f:
 from style_bert_vits2.nlp import bert_models
 from style_bert_vits2.constants import Languages
 
-bert_models.load_model(Languages.ZH, "microsoft/deberta-v3-large")
-bert_models.load_tokenizer(Languages.ZH, "microsoft/deberta-v3-large")
+bert_models.load_model(Languages.ZH, "hfl/chinese-roberta-wwm-ext-large")
+bert_models.load_tokenizer(Languages.ZH, "hfl/chinese-roberta-wwm-ext-large")
 
 # save model files to model_assets dir
 from pathlib import Path
 from huggingface_hub import hf_hub_download
 
-model_file = "rikka_botan_english.safetensors"
+model_file = "rikka_botan_chinese.safetensors"
 config_file = "config.json"
 style_file = "style_vectors.npy"
 
 for file in [model_file, config_file, style_file]:
     print(file)
     hf_hub_download(
-        "RikkaBotan/style_bert_vits2_english_original",
+        "RikkaBotan/style_bert_vits2_chinese_original",
         file,
         local_dir="model_assets"
     )
@@ -226,7 +226,7 @@ model = TTSModel(
     device="cuda"  # If you cannot use cuda, please input cpu
 )
 
-# Please input the English text
+# Please input the Chinese text
 from IPython.display import Audio, display
 
 sr, audio = model.infer(text="please write text here")