Arrangements will depend heavily on the country the baby was born.
You will need an Immigration Solicitor with a specialising in Surrogacy  to advise and assist in developing a strategy to ensure the baby can return to the UK post birth asap.
We will advise you on the best option for you and can provide guidance on how to secure comprehensive immigration advice.