AI-based model