Load_learner from fastai is predicting a subset of whole class. Not sure where I am doing wrong