dagan-demo / app.py
Matt Goyder
initial commit
0c47772
raw
history blame
720 Bytes
import os
import gradio as gr
import subprocess
title = "DaGAN Demo"
description = "reconstruction and animation demos for DaGAN"
def inference(mode, img, video):
os.makedirs('tmp', exist_ok=True)
subprocess.run("echo 'hi'")
"""
os.system("python run.py --source_image {} --driving_video
"""
gr.Interface(
inference,
[
gr.Dropdown(["reconstruction", "animation"]),
gr.inputs.Image(type="filepath", label="Source (only used if mode is reconstruction)"),
gr.inputs.Image(type="mp4", label="Driving Video")
],
outputs=None,
title=title,
description=description,
theme="huggingface",
allow_flagging=False).launch(debug=False,enable_queue=True)