Machine learning for content creation