import os, sys import tempfile import gradio as gr from src.gradio_demo import SadTalker from src.utils.text2speech import TTSTalker from huggingface_hub import snapshot_download from src.utils.text2speech import TTSTalkerPlayHT def get_source_image(image): return image def download_model(): REPO_ID = 'vinthony/SadTalker' snapshot_download(repo_id=REPO_ID, local_dir='./checkpoints', local_dir_use_symlinks=True) def sadtalker_demo(): # 这部分的逻辑是在我的笔记本上本地运行和在服务器上运行的时候，模型的路径不一样，所以需要做一下判断 import platform if platform.system() != 'Windows': download_model() sad_talker = SadTalker( lazy_load=True, checkpoint_path='./checkpoints' if platform.system() == 'Linux' else 'checkpoints_win', ) tts_talker = TTSTalker() tts_talker_ht = TTSTalkerPlayHT() with gr.Blocks(analytics_enabled=False) as sadtalker_interface: gr.Markdown("

😭 SadTalker: Learning Realistic 3D Motion Coefficients for Stylized Audio-Driven Single Image Talking Face Animation (CVPR 2023)

\ Arxiv       \ Homepage       \ Github
") gr.Markdown(""" You may duplicate the space and upgrade to GPU in settings for better performance and faster inference without waiting in the queue. Duplicate Space \
Alternatively, try our GitHub code on your own GPU. \ """) with gr.Row().style(equal_height=False): with gr.Column(variant='panel'): with gr.Tabs(elem_id="sadtalker_source_image"): with gr.TabItem('Upload image'): with gr.Row(): source_image = gr.Image(label="Source image", source="upload", type="filepath").style(height=256,width=256) # with gr.Tabs(elem_id="sadtalker_driven_audio"): # with gr.TabItem('Upload or Generating from TTS'): # with gr.Column(variant='panel'): # driven_audio = gr.Audio(label="Input audio(.wav/.mp3)", source="upload", type="filepath") # with gr.Column(variant='panel'): # input_text = gr.Textbox(label="Generating audio from text", lines=5, placeholder="Alternatively, you can 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]) # ht TTS with gr.Tabs(elem_id="sadtalker_driven_audio"): with gr.TabItem('Play.ht: Upload OR TTS'): with gr.Column(variant='panel'): gr.Markdown("find more info here: https://playht.github.io/api-docs-generator/#standard-api-voices") driven_audio = gr.Audio(label="Input audio", source="upload", type="filepath") ht_user_id = gr.Textbox(label="Play.ht User ID", lines=1, placeholder="(required)", value='96tPb0H2cXbobV9u8iLVGyJPUPc2') ht_auth_key = gr.Textbox(label="Play.ht Auth Key", lines=1, placeholder="(required)",value='f35fc9d7ce0549a88f6cdc15ec860b6e') ht_voice = gr.Textbox(label="Play.ht Voice Type, see section in API doc", lines=1, placeholder="(required)",value='en-US-MichelleNeural') 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_ht.test, inputs=[input_text, ht_user_id, ht_auth_key,ht_voice], outputs=[driven_audio]) # origin TTS with gr.Tabs(elem_id="sadtalker_driven_audio"): with gr.TabItem('Origin: Upload OR TTS'): with gr.Column(variant='panel'): driven_audio = gr.Audio(label="Input audio", source="upload", type="filepath") 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]) with gr.Column(variant='panel'): with gr.Tabs(elem_id="sadtalker_checkbox"): with gr.TabItem('Settings'): with gr.Column(variant='panel'): preprocess_type = gr.Radio(['crop','resize','full'], value='crop', label='preprocess', info="How to handle input image?") is_still_mode = gr.Checkbox(label="w/ Still Mode (fewer hand motion, works with preprocess `full`)") enhancer = gr.Checkbox(label="w/ GFPGAN as Face enhancer") submit = gr.Button('Generate', elem_id="sadtalker_generate", variant='primary') with gr.Tabs(elem_id="sadtalker_genearted"): gen_video = gr.Video(label="Generated video", format="mp4").style(width=256) with gr.Row(): examples = [ [ 'examples/source_image/full_body_1.png', 'examples/driven_audio/bus_chinese.wav', 'crop', True, False ], [ 'examples/source_image/full_body_2.png', 'examples/driven_audio/japanese.wav', 'crop', False, False ], [ 'examples/source_image/full3.png', 'examples/driven_audio/deyu.wav', 'crop', False, True ], [ 'examples/source_image/full4.jpeg', 'examples/driven_audio/eluosi.wav', 'full', False, True ], [ 'examples/source_image/full4.jpeg', 'examples/driven_audio/imagine.wav', 'full', True, True ], [ 'examples/source_image/full_body_1.png', 'examples/driven_audio/bus_chinese.wav', 'full', True, False ], [ 'examples/source_image/art_13.png', 'examples/driven_audio/fayu.wav', 'resize', True, False ], [ 'examples/source_image/art_5.png', 'examples/driven_audio/chinese_news.wav', 'resize', False, False ], [ 'examples/source_image/art_5.png', 'examples/driven_audio/RD_Radio31_000.wav', 'resize', True, True ], ] gr.Examples(examples=examples, inputs=[ source_image, driven_audio, preprocess_type, is_still_mode, enhancer, # ht_user_id, # ht_auth_key ], outputs=[gen_video], fn=sad_talker.test, cache_examples=False # cache_examples=os.getenv('SYSTEM') == 'spaces' ) # submit.click( fn=sad_talker.test, inputs=[source_image, driven_audio, preprocess_type, is_still_mode, enhancer], outputs=[gen_video] ) return sadtalker_interface if __name__ == "__main__": demo = sadtalker_demo() demo.queue(max_size=10) demo.launch(debug=True)