use burn_onnx::ModelGen;

fn main() {
    // Generate the model code from the ONNX file.
    // Model weights are stored in .burnpack format and loaded at runtime.
    ModelGen::new()
        .input("src/model/mnist.onnx")
        .out_dir("model/")
        .run_from_script();
}
