Spaces:
Runtime error
Runtime error
import spaces | |
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]) | |