diff --git a/core/config.py b/core/config.py index fb69913..df1c3ca 100644 --- a/core/config.py +++ b/core/config.py @@ -7,4 +7,7 @@ face_analyser.prepare(ctx_id=0, det_size=(640, 640)) def get_face(img_data): analysed = face_analyser.get(img_data) - return sorted(analysed, key=lambda x: x.bbox[0])[0] + try: + return sorted(analysed, key=lambda x: x.bbox[0])[0] + except IndexError: + return None