A two-layer depth-averaged approach to describe the regime stratification in collapses of dry granular columns