code: def transform_fn(net, data, input_content_type, output_content_type):
      
      # we can use content types to vary input/output handling, but
      # here we just assume json for both
      parsed = json.loads(data)
      nda = mx.nd.array(parsed)
      output = net(nda)
      prediction = mx.nd.argmax(output, axis=1)
      response_body = json.dumps(prediction.asnumpy().tolist())
      return response_body, output_content_type
docs: Transform a request using the Gluon model. Called once per request.
  
      :param net: The Gluon model.
      :param data: The request payload.
      :param input_content_type: The request content type.
      :param output_content_type: The (desired) response content type.
      :return: response payload and content type.