Pipeline update & tests
Image classification models that have less than 5 labels currently cannot run with the pipeline defaults as it uses a top_k of 5 by default. This puts a limit on the top_k so that it maxes out to the number of labels of the model.