Serving models with number of outputs less than 5

[sergey-serebryakov]

Hi!
I think, in current implementation the engine cannot server models if they have less than five outputs. The classify method accepts as one of the input parameters N which is the number of top classifications to return. The Argmax function fails in this case.

[flx42]

You have to edit the code in this case, to modify the Classify call.