Here is the suggested design to improve the loading effect in AI Categories
It uses the same colors as the "Generate by AI" button while applying black and white for other loading effects across the platform.