| import os | |
| os.system("hub install wav2lip==1.0.0") | |
| import gradio as gr | |
| import paddlehub as hub | |
| from pathlib import Path | |
| module = hub.Module(name="wav2lip") | |
| def inference(image,audio): | |
| face_input_path = image | |
| audio_input_path = audio | |
| module.wav2lip_transfer(face=face_input_path, audio=audio_input_path, output_dir='./transfer_result/', use_gpu=False) | |
| return './transfer_result/'+Path(image.name).stem+".mp4" | |
| title = "Wav2lip" | |
| description = "Gradio demo for Wav2lip. To use it, simply upload your image, or click one of the examples to load them. Read more at the links below." | |
| article = "<p style='text-align: center'><a href='https://github.com/jantic/DeOldify' target='_blank'>Github Repo</a></p>" | |
| examples=[['lunch.jpeg']] | |
| iface = gr.Interface(inference, [gr.inputs.Image(type="filepath"),gr.inputs.Audio(type="filepath")], | |
| outputs=gr.outputs.Video(label="Output Video"),examples=examples,enable_queue=True,title=title,article=article,description=description) | |
| iface.launch() |