er1t0 commited on
Commit
3d78f56
1 Parent(s): 2dbbf29

initial setiup

Browse files
Files changed (2) hide show
  1. app.py +33 -0
  2. requirements.txt +5 -0
app.py ADDED
@@ -0,0 +1,33 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ import gradio as gr
2
+ from deepface import DeepFace
3
+ import cv2
4
+ import numpy as np
5
+
6
+ def analyze_frame(frame):
7
+ result = DeepFace.analyze(frame, actions=['gender'], enforce_detection=False)
8
+ if result:
9
+ gender = result[0]['gender']
10
+ else:
11
+ gender = "No face detected"
12
+ return frame, f"Gender: {gender}"
13
+
14
+ def video_feed():
15
+ camera = cv2.VideoCapture(0)
16
+ while True:
17
+ ret, frame = camera.read()
18
+ if not ret:
19
+ break
20
+ yield frame
21
+ camera.release()
22
+
23
+ with gr.Blocks() as demo:
24
+ gr.Markdown("# Real-time Gender Detection")
25
+
26
+ with gr.Row():
27
+ video_stream = gr.Video(source=video_feed, streaming=True)
28
+ result_image = gr.Image()
29
+
30
+ video_stream.stream(fn=analyze_frame, inputs=video_stream, outputs=result_image)
31
+
32
+ if __name__ == "__main__":
33
+ demo.launch()
requirements.txt ADDED
@@ -0,0 +1,5 @@
 
 
 
 
 
 
1
+ deepface
2
+ opencv-python
3
+ numpy
4
+ tf-keras
5
+ gradio