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='.', use_gpu=True)  
    return "result.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=[['monatest.jpeg','game.wav']]
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(debug=True)