VITS/app.py

from GPT_SoVITS import *
import gradio as gr

GPT_SoVITS_inference = GPT_SoVITS()
gpt_path = "GPT_SoVITS/pretrained_models/s1bert25hz-2kh-longer-epoch=68e-step=50232.ckpt"
sovits_path = "GPT_SoVITS/pretrained_models/s2G488k.pth"
GPT_SoVITS_inference.load_model(gpt_path, sovits_path)

with gr.Blocks(title = "GPT-SoVITS") as demo:
 with gr.Row():
 gr.HTML("<center><h1>GPT-SoVITS WebUI</h1></center>")
 gr.Markdown(value="*请上传并填写参考信息")
 with gr.Row():
 inp_ref = gr.Audio(label="请上传3~10秒内参考音频，超过会报错！", sources=["microphone", "upload"], type="filepath")
 prompt_text = gr.Textbox(label="参考音频的文本", value="")
 prompt_language = gr.Dropdown(
 label="参考音频的语种", choices=["中文", "英文", "日文"], value="中文"
 )
 gr.Markdown(value="*请填写需要合成的目标文本。中英混合选中文，日英混合选日文，中日混合暂不支持，非目标语言文本自动遗弃。")
 with gr.Row():
 text = gr.Textbox(label="需要合成的文本", value="")
 text_language = gr.Dropdown(
 label="需要合成的语种", choices=["中文", "英文", "日文", "中英混合", "日英混合", "多语种混合"], value="中文"
 )
 how_to_cut = gr.Radio(
 label="怎么切",
 choices=["不切", "凑四句一切", "凑50字一切", "按中文句号。切", "按英文句号.切", "按标点符号切" ],
 value="凑四句一切",
 interactive=True,
 )
 inference_button = gr.Button("合成语音", variant="primary")
 output = gr.Audio(label="输出的语音")

 inference_button.click(
 GPT_SoVITS_inference.predict,
 [inp_ref, prompt_text, prompt_language, text, text_language, how_to_cut],
 [output],
 )
 gr.Markdown(value="文本切分工具。太长的文本合成出来效果不一定好，所以太长建议先切。合成会根据文本的换行分开合成再拼起来。")
 with gr.Row():
 text_inp = gr.Textbox(label="需要合成的切分前文本", value="")
 button1 = gr.Button("凑四句一切", variant="primary")
 button2 = gr.Button("凑50字一切", variant="primary")
 button3 = gr.Button("按中文句号。切", variant="primary")
 button4 = gr.Button("按英文句号.切", variant="primary")
 button5 = gr.Button("按标点符号切", variant="primary")
 text_opt = gr.Textbox(label="切分后文本", value="")
 button1.click(cut1, [text_inp], [text_opt])
 button2.click(cut2, [text_inp], [text_opt])
 button3.click(cut3, [text_inp], [text_opt])
 button4.click(cut4, [text_inp], [text_opt])
 button5.click(cut5, [text_inp], [text_opt])
 gr.Markdown(value="后续将支持混合语种编码文本输入。")

demo.launch()