Storing a model in ONNX format