Spaces:
Runtime error
Runtime error
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]) | |