Hy @Bisma_Akram



 Bisma_Akram:

 for label in classes:

      damage_labels.append(damage_dict[label])



when you iterate over classes, your classes list contains the anomly either check this column or remove it from your dataset. It’s better if you use exception handling here.