YOLOv5 ptl model inference results out of bounds of classes array

I have a yolov5 model that I converted into ptl using the code below