code_example.swift

private func stylizeImage(cgImage: CGImage, model: MLModel) -> CGImage {
    
    // size can change here if you want, remember to run right sizes in the fst evaluating script
    let input = StyleTransferInput(input: pixelBuffer(cgImage: cgImage, width: 883, height: 720))

    // model.prediction will run the style model on input image
    let outFeatures = try! model.prediction(from: input)
  
    // we get the image buffer after
    let output = outFeatures.featureValue(for: "add_37__0")!.imageBufferValue!
  
    // remaining code to convert image buffer here .....
}