I have a MLModel which is of type neural network. It accepts the input as MLMultiArray 1x3 and outputs the MLMultiArray. This is for the words prediction based on a single o