let stylizationModel;

​

function preload() {

  inputImage = loadImage('Kitten_0.png'); // Replace 'input.jpg' with your input image file

}

​

function setup() {

  createCanvas(inputImage.width, inputImage.height);

  image(inputImage, 0, 0);

​

  // Load the pre-trained style transfer model

  ml5.styleTransfer('arbitrary', modelLoaded);

}

​

function modelLoaded(model) {

  stylizationModel = model;

  console.log('Style transfer model loaded.');

}

​

function mouseReleased() {

  if (stylizationModel) {

    // Perform style transfer

    stylizationModel.transfer(inputImage, function (err, stylizedImage) {

      if (err) {

        console.error('Style transfer error:', err);

      } else {

        // Display the stylized image on the canvas

        image(stylizedImage, 0, 0);

      }

    });

  }

}

​