CosyVoice-300M / css /preset.py
ljy266987
add lfs
12bfd03
raw
history blame
2.29 kB
import random
import gradio as gr
from css.utils import *
# 预置语音生成
def preset():
sound_choices = ['中文女', '中文男', '英文女', '英文男', '日语男', '粤语女', '韩语女']
def random_seed():
return random.randint(1, 100000000)
def generate_audio(_sound_radio, _synthetic_input_textbox, _seed):
print(_sound_radio, _synthetic_input_textbox, _seed)
if _synthetic_input_textbox == '':
gr.Warning('合成文本为空,您是否忘记输入合成文本?')
return (target_sr, default_data)
set_all_random_seed(_seed)
if use_instruct(_synthetic_input_textbox):
model = cosyvoice_instruct
else:
model = cosyvoice_sft
output = model.inference_sft(_synthetic_input_textbox, _sound_radio)
audio_data = postprocess(output['tts_speech']).numpy().flatten()
return (target_sr, audio_data)
with gr.Column():
sound_radio = gr.Radio(choices=sound_choices,
value=sound_choices[0],
label="选择预置音色")
with gr.Column():
synthetic_input_textbox = gr.Textbox(label="输入合成文本")
gr.Examples(
label="示例文本",
examples=example_tts_text,
inputs=[synthetic_input_textbox])
with gr.Accordion(label="随机种子"):
with gr.Row():
with gr.Column(scale=1, min_width=180):
seed_button = gr.Button(value="\U0001F3B2 随机换一换",
elem_classes="full-height")
with gr.Column(scale=10):
seed = gr.Number(show_label=False,
value=0,
container=False,
elem_classes="full-height")
with gr.Column():
generate_button = gr.Button("生成音频", variant="primary", size="lg")
with gr.Column():
output_audio = gr.Audio(label="合成音频")
seed_button.click(fn=random_seed, outputs=[seed])
generate_button.click(fn=generate_audio,
inputs=[sound_radio, synthetic_input_textbox, seed],
outputs=[output_audio])