Decode_predictions wrong shape

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.

I think Raymond answered your question on the other thread where you asked about this.