Spaces:
Sleeping
Sleeping
Update app.py
Browse files
app.py
CHANGED
@@ -24,23 +24,23 @@ if os.path.exists("./gweight.txt"):
|
|
24 |
"gpt_path", gweight_data)
|
25 |
else:
|
26 |
gpt_path = os.environ.get(
|
27 |
-
"gpt_path", "
|
28 |
|
29 |
if os.path.exists("./sweight.txt"):
|
30 |
with open("./sweight.txt", 'r', encoding="utf-8") as file:
|
31 |
sweight_data = file.read()
|
32 |
sovits_path = os.environ.get("sovits_path", sweight_data)
|
33 |
else:
|
34 |
-
sovits_path = os.environ.get("sovits_path", "
|
35 |
# gpt_path = os.environ.get(
|
36 |
# "gpt_path", "pretrained_models/s1bert25hz-2kh-longer-epoch=68e-step=50232.ckpt"
|
37 |
# )
|
38 |
# sovits_path = os.environ.get("sovits_path", "pretrained_models/s2G488k.pth")
|
39 |
cnhubert_base_path = os.environ.get(
|
40 |
-
"cnhubert_base_path", "
|
41 |
)
|
42 |
bert_path = os.environ.get(
|
43 |
-
"bert_path", "
|
44 |
)
|
45 |
infer_ttswebui = os.environ.get("infer_ttswebui", 9872)
|
46 |
infer_ttswebui = int(infer_ttswebui)
|
@@ -584,8 +584,8 @@ def change_choices():
|
|
584 |
return {"choices": sorted(SoVITS_names, key=custom_sort_key), "__type__": "update"}, {"choices": sorted(GPT_names, key=custom_sort_key), "__type__": "update"}
|
585 |
|
586 |
|
587 |
-
pretrained_sovits_name = "
|
588 |
-
pretrained_gpt_name = "
|
589 |
SoVITS_weight_root = "SoVITS_weights"
|
590 |
GPT_weight_root = "GPT_weights"
|
591 |
os.makedirs(SoVITS_weight_root, exist_ok=True)
|
|
|
24 |
"gpt_path", gweight_data)
|
25 |
else:
|
26 |
gpt_path = os.environ.get(
|
27 |
+
"gpt_path", "pretrained_models/hutao-e15.ckpt")
|
28 |
|
29 |
if os.path.exists("./sweight.txt"):
|
30 |
with open("./sweight.txt", 'r', encoding="utf-8") as file:
|
31 |
sweight_data = file.read()
|
32 |
sovits_path = os.environ.get("sovits_path", sweight_data)
|
33 |
else:
|
34 |
+
sovits_path = os.environ.get("sovits_path", "pretrained_models/hutao_e12_s1740.pth")
|
35 |
# gpt_path = os.environ.get(
|
36 |
# "gpt_path", "pretrained_models/s1bert25hz-2kh-longer-epoch=68e-step=50232.ckpt"
|
37 |
# )
|
38 |
# sovits_path = os.environ.get("sovits_path", "pretrained_models/s2G488k.pth")
|
39 |
cnhubert_base_path = os.environ.get(
|
40 |
+
"cnhubert_base_path", "pretrained_models/chinese-hubert-base"
|
41 |
)
|
42 |
bert_path = os.environ.get(
|
43 |
+
"bert_path", "pretrained_models/chinese-roberta-wwm-ext-large"
|
44 |
)
|
45 |
infer_ttswebui = os.environ.get("infer_ttswebui", 9872)
|
46 |
infer_ttswebui = int(infer_ttswebui)
|
|
|
584 |
return {"choices": sorted(SoVITS_names, key=custom_sort_key), "__type__": "update"}, {"choices": sorted(GPT_names, key=custom_sort_key), "__type__": "update"}
|
585 |
|
586 |
|
587 |
+
pretrained_sovits_name = "pretrained_models/hutao-e15.ckpt"
|
588 |
+
pretrained_gpt_name = "pretrained_models/hutao_e12_s1740.pth"
|
589 |
SoVITS_weight_root = "SoVITS_weights"
|
590 |
GPT_weight_root = "GPT_weights"
|
591 |
os.makedirs(SoVITS_weight_root, exist_ok=True)
|