Models of mass segregation at the Galactic centre