Core ML Swift API correction
