mmesa-gpu-gitex / tabs /FACS_analysis.py
vitorcalvi's picture
1
da21442
import gradio as gr
from app.app_utils import preprocess_frame_and_predict_aus
def create_facs_analysis_tab(pth_model_static, pth_model_dynamic, cam):
def analyze_facs(frame):
cur_face, au_intensities, heatmap = preprocess_frame_and_predict_aus(frame)
return cur_face, heatmap, au_intensities
with gr.Row():
with gr.Column():
input_image = gr.Image(type="numpy", label="Upload a Frame")
analyze_button = gr.Button("Analyze")
with gr.Column():
output_face = gr.Image(type="numpy", label="Detected Face")
output_heatmap = gr.Image(type="numpy", label="Heatmap")
output_aus = gr.Label(label="AU Intensities")
analyze_button.click(analyze_facs, inputs=[input_image], outputs=[output_face, output_heatmap, output_aus])
return gr.Row([input_image, analyze_button, output_face, output_heatmap, output_aus])