African Union mission in Somalia says al-Shabab rebels are on the run.