iOS Question MLKit Selfie Segmentation

-(void (^)(id, id)) createBlock:(CVPixelBufferRef)pixelBuffer {
    void (^block)(id result, id error) = ^void (MLKSegmentationMask* mask, id error) {
        if (error != nil || mask == nil) {
            [B4I shared].lastError = error;
            [self.bi raiseUIEvent:nil event:@"process_result::" params:@[@(false), [NSNull null]]];
        } else {
            // Get mask dimensions and pixel buffer
            CVPixelBufferRef maskPixelBuffer = mask.buffer;
            CIImage *ciMaskImage = [CIImage imageWithCVPixelBuffer:maskPixelBuffer];
           
            // Ensure mask is in grayscale
            ciMaskImage = [ciMaskImage imageByApplyingFilter:@"CIColorMatrix"
                                   withInputParameters:@{
                                       @"inputRVector": [CIVector vectorWithX:0 Y:0 Z:0 W:1],
                                       @"inputGVector": [CIVector vectorWithX:0 Y:0 Z:0 W:1],
                                       @"inputBVector": [CIVector vectorWithX:0 Y:0 Z:0 W:1],
                                       @"inputAVector": [CIVector vectorWithX:0 Y:0 Z:0 W:1]
                                   }];
           
            CIContext *temporaryContext = [CIContext contextWithOptions:nil];
           
            // Convert the original pixel buffer to CIImage
            CIImage *originalImage = [CIImage imageWithCVPixelBuffer:pixelBuffer];

            // Ensure original image and mask have matching dimensions
            if (CVPixelBufferGetWidth(maskPixelBuffer) != CVPixelBufferGetWidth(pixelBuffer) ||
                CVPixelBufferGetHeight(maskPixelBuffer) != CVPixelBufferGetHeight(pixelBuffer)) {
                // Resize original image to match mask dimensions if necessary
                originalImage = [originalImage imageByApplyingFilter:@"CILanczosScaleTransform"
                                                withInputParameters:@{
                                                    kCIInputScaleKey: @(CVPixelBufferGetWidth(maskPixelBuffer) / CVPixelBufferGetWidth(pixelBuffer)),
                                                    kCIInputAspectRatioKey: @(1.0)
                                                }];
            }

            // Blend the mask and the original image
            CIImage *maskedImage = [originalImage imageByApplyingFilter:@"CIBlendWithAlphaMask"
                                    withInputParameters:@{
                                        kCIInputMaskImageKey: ciMaskImage
                                    }];
           
            // Convert the result to a UIImage
            CGImageRef videoImage = [temporaryContext createCGImage:maskedImage
                    fromRect:CGRectMake(0, 0,
                    CVPixelBufferGetWidth(maskPixelBuffer),
                    CVPixelBufferGetHeight(maskPixelBuffer))];
                   
            UIImage *uiImage = [UIImage imageWithCGImage:videoImage];
            CGImageRelease(videoImage);
           
            // Raise event with the processed image
            [self.bi raiseUIEvent:nil event:@"process_result::" params:@[@(true), uiImage]];
        }
    };
    return block;
}