ERROR WHILE RUNNING train_bottleneck.py ???

[parthmehta15]

I am trying to train the model for 2 classes but I am getting this error:

AssertionError: class id must be less than num_classes

[pythonlessons]

Fix your classes file