Modelling the HI halo of the Milky Way