goldpulpy's picture
Update app.py
5bd9a13
raw
history blame
No virus
669 Bytes
import random
import subprocess
import os
import gradio
import gradio as gr
import requests
current_dir = os.path.dirname(os.path.abspath(__file__))
def predict(video, audio):
print(f"Processing video {video} and audio {audio}")
output_file = f"{current_dir}/results/{random.randint(10,100000)}.mp4"
command = ["python", "inference.py", "--face", video,
"--audio", audio, "--outfile", output_file]
subprocess.run(command, check=True)
return output_file
demo = gr.Interface(fn=predict, inputs=[gradio.Video(label='Source image/video'),gradio.Audio(type='filepath', label='Target Audio')], outputs=[gradio.Video()])
demo.launch()