The model.predict output is either [[1.0.0.]], [[0.1.0.]], [[0.0.1.]] depending on the category of the image. currently struggling to get the category name as output instead
