paresh95
PS|Modularised face proportions
4ff0905
raw
history blame
1.17 kB
import gradio as gr
from src.face_texture import GetFaceTexture
from src.face_symmetry import GetFaceSymmetry
from src.face_demographics import GetFaceDemographics
from src.face_proportions import GetFaceProportions
def combined_fn(input_image):
texture_results = GetFaceTexture().main(input_image)
symmetry_results = GetFaceSymmetry().main(input_image)
demographics_results = GetFaceDemographics().main(input_image)
proportion_results = GetFaceProportions().main(input_image)
return (*texture_results, *symmetry_results, demographics_results, *proportion_results)
iface = gr.Interface(
fn=combined_fn,
inputs=gr.inputs.Image(type="pil"),
outputs=[
gr.outputs.Image(type="pil"), # From GetFaceTexture
gr.outputs.Image(type="pil"), # From GetFaceTexture
"text", # From GetFaceTexture
gr.outputs.Image(type="pil"), # From GetFaceSymmetry
"text", # From GetFaceSymmetry
"text", # From GetFaceDemographics
"text", # From GetFaceProportions
"text", # From GetFaceProportions
gr.outputs.Image(type="pil"), # From GetFaceProportions
],
)
iface.launch()