File size: 854 Bytes
3d78f56
 
 
 
 
 
51de831
 
 
3d78f56
 
 
 
51de831
3d78f56
51de831
 
 
 
 
3d78f56
51de831
 
 
 
 
 
3d78f56
 
51de831
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
import gradio as gr
from deepface import DeepFace
import cv2
import numpy as np

def analyze_frame(frame):
    # Convert frame from BGR to RGB format for DeepFace
    frame_rgb = cv2.cvtColor(frame, cv2.COLOR_BGR2RGB)
    result = DeepFace.analyze(frame_rgb, actions=['gender'], enforce_detection=False)
    if result:
        gender = result[0]['gender']
    else:
        gender = "No face detected"
    return frame, gender

def process_webcam_image(image):
    # Convert the image to a format suitable for DeepFace
    image = cv2.cvtColor(np.array(image), cv2.COLOR_RGB2BGR)
    frame, gender = analyze_frame(image)
    return frame, gender

demo = gr.Interface(
    fn=process_webcam_image, 
    inputs=gr.Image(source="webcam", streaming=True), 
    outputs=[gr.Image(), gr.Textbox()],
    live=True
)

if __name__ == "__main__":
    demo.launch()