blackteay/GPT-SoVITS
1
0
Fork 0
mirror of https://github.com/RVC-Boss/GPT-SoVITS.git synced 2025-10-06 22:50:00 +08:00
Code Issues Packages Projects Releases Wiki Activity

Update webui.py

Browse Source 
添加faster-whisper转写入口
This commit is contained in:
刘悦 2024-01-28 15:43:16 +08:00 committed by GitHub
parent 0098f8d742
commit 18c1730ef2
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 19 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

19
webui.py
View File

@ -189,6 +189,23 @@ def change_tts_inference(if_tts,bert_path,cnhubert_base_path,gpu_number,gpt_path
p_tts_inference=None
yield i18n("TTS推理进程已关闭")
def open_asr_whisper(asr_inp_dir,model_name):
global p_asr
if(p_asr==None):
#cmd = '"%s" tools/damo_asr/cmd-asr.py "%s"'%(python_exec,asr_inp_dir)
cmd = '"%s" tools/damo_asr/faster-whisper-asr.py "%s" "%s"'%(python_exec,asr_inp_dir,model_name)
yield "ASR任务开启：%s"%cmd,{"__type__":"update","visible":False},{"__type__":"update","visible":True}
print(cmd)
p_asr = Popen(cmd, shell=True)
p_asr.wait()
p_asr=None
yield "ASR任务完成",{"__type__":"update","visible":True},{"__type__":"update","visible":False}
else:
yield "已有正在进行的ASR任务，需先终止才能开启下一次任务",{"__type__":"update","visible":False},{"__type__":"update","visible":True}
def open_asr(asr_inp_dir):
global p_asr
@ -650,6 +667,7 @@ with gr.Blocks(title="GPT-SoVITS WebUI") as app:
slicer_info = gr.Textbox(label=i18n("语音切割进程输出信息"))
gr.Markdown(value=i18n("0c-中文批量离线ASR工具"))
with gr.Row():
open_asr_button_whisper = gr.Button(i18n("开启Faster-Whisper离线ASR"), variant="primary",visible=True)
open_asr_button = gr.Button(i18n("开启离线批量ASR"), variant="primary",visible=True)
close_asr_button = gr.Button(i18n("终止ASR进程"), variant="primary",visible=False)
asr_inp_dir = gr.Textbox(
@ -670,6 +688,7 @@ with gr.Blocks(title="GPT-SoVITS WebUI") as app:
if_label.change(change_label, [if_label,path_list], [label_info])
if_uvr5.change(change_uvr5, [if_uvr5], [uvr5_info])
open_asr_button.click(open_asr, [asr_inp_dir], [asr_info,open_asr_button,close_asr_button])
open_asr_button_whisper.click(open_asr_whisper, [asr_inp_dir,model_version], [asr_info,open_asr_button,close_asr_button])
close_asr_button.click(close_asr, [], [asr_info,open_asr_button,close_asr_button])
open_slicer_button.click(open_slice, [slice_inp_path,slice_opt_root,threshold,min_length,min_interval,hop_size,max_sil_kept,_max,alpha,n_process], [slicer_info,open_slicer_button,close_slicer_button])
close_slicer_button.click(close_slice, [], [slicer_info,open_slicer_button,close_slicer_button])