From 1b01a4489f19969a43b8a298d8eb365d18ea275b Mon Sep 17 00:00:00 2001 From: Max Jiang Date: Tue, 5 Dec 2023 12:39:49 +1100 Subject: [PATCH] fix-gradio-audio --- app_sadtalker.py | 2 +- src/utils/text2speech.py | 4 ++-- 2 files changed, 3 insertions(+), 3 deletions(-) diff --git a/app_sadtalker.py b/app_sadtalker.py index 1401a600..aca462f0 100644 --- a/app_sadtalker.py +++ b/app_sadtalker.py @@ -50,7 +50,7 @@ def sadtalker_demo(checkpoint_path='checkpoints', config_path='src/config', warp with gr.Column(variant='panel'): input_text = gr.Textbox(label="Generating audio from text", lines=5, placeholder="please enter some text here, we genreate the audio from text using @Coqui.ai TTS.") tts = gr.Button('Generate audio',elem_id="sadtalker_audio_generate", variant='primary') - tts.click(fn=tts_talker.test, inputs=[input_text], outputs=[driven_audio]) + tts.click(fn=tts_talker.test, inputs=[input_text, driven_audio], outputs=[driven_audio]) with gr.Column(variant='panel'): with gr.Tabs(elem_id="sadtalker_checkbox"): diff --git a/src/utils/text2speech.py b/src/utils/text2speech.py index a9cd9f58..a51a7d34 100644 --- a/src/utils/text2speech.py +++ b/src/utils/text2speech.py @@ -8,13 +8,13 @@ def __init__(self) -> None: model_name = TTS().list_models()[0] self.tts = TTS(model_name) - def test(self, text, language='en'): + def test(self, text, audio, language='en'): tempf = tempfile.NamedTemporaryFile( delete = False, suffix = ('.'+'wav'), ) - self.tts.tts_to_file(text, speaker=self.tts.speakers[0], language=language, file_path=tempf.name) + self.tts.tts_to_file(text, speaker_wav=audio, language=language, file_path=tempf.name) return tempf.name