AI-based quality inspection