Added Face detection

app.py

@@ -30,8 +30,6 @@ def detect(img_raw):
             dets = face_detection(img)
         dets = [FaceDetectorResult(o) for o in dets[0]]
 
-
-
         img_vis = img_raw.copy()
 
         vis_threshold = 0.8
@@ -40,15 +38,19 @@ def detect(img_raw):
             if b.score < vis_threshold:
                 continue
 
-            # draw face bounding box
+            # Draw face bounding box
             img_vis = cv2.rectangle(img_vis, b.top_left.int().tolist(), b.bottom_right.int().tolist(), (0, 255, 0), 4)
-
+            # Draw Keypoints
+            img_vis = draw_keypoint(img_vis, b, FaceKeypoint.EYE_LEFT)
+            img_vis = draw_keypoint(img_vis, b, FaceKeypoint.EYE_RIGHT)
+            img_vis = draw_keypoint(img_vis, b, FaceKeypoint.NOSE)
+            img_vis = draw_keypoint(img_vis, b, FaceKeypoint.MOUTH_LEFT)
+            img_vis = draw_keypoint(img_vis, b, FaceKeypoint.MOUTH_RIGHT)
 
         return img_vis
 
 
 
-
 title = "Kornia Face Detection"
 description = "<p style='text-align: center'>This is a Gradio demo for Kornia's Face Detection.</p><p style='text-align: center'>To use it, simply upload your image, or click one of the examples to load them</p>"
 article = "<p style='text-align: center'><a href='https://kornia.readthedocs.io/en/latest/' target='_blank'>Kornia Docs</a> | <a href='https://github.com/kornia/kornia' target='_blank'>Kornia Github Repo</a> | <a href='https://kornia.readthedocs.io/en/latest/applications/face_detection.html' target='_blank'>Kornia Face Detection Tutorial</a></p>"