Update `main.cpp` so that it instantiates the model

[syclik]

Let's get a basic model instantiated and show that we can get the gradient out of it.