Face Reorganization after prediction try to face index of predict value then getting index 3 is out of bounds for axis 0 with size 2

Question

In the following code I get the error

index 3 is out of bounds for axis 0 with size 2

for the statement pred[0][3]. What am