I got this error:
ValueError: decode_predictions
expects a batch of predictions (i.e. a 2D array of shape (samples, 1000)). Found array with shape: (2, 108)
I am confused, because the array is of the required shape. Samples = 2, classes =108. My output is in the form of an array of one_hots, with zeros in every index but the class identified.