label:bug detect_faces() not working

[agarciadelasanta]

I tried to run the code and i got the following error:
IndexError Traceback (most recent call last)
in ()
1 detector = MTCNN()
----> 2 detector.detect_faces(img)

2 frames
/content/mtcnn_tflite/MTCNN.py in __generate_bounding_box(imap, reg, scale, t)
139 dx1 = np.transpose(reg[:, :, 0])
140 dy1 = np.transpose(reg[:, :, 1])
--> 141 dx2 = np.transpose(reg[:, :, 2])
142 dy2 = np.transpose(reg[:, :, 3])
143

IndexError: index 2 is out of bounds for axis 2 with size 2