let button;

let txt;

let label;

let score;

​

async function preload() {

  // this loads distilbert... for a sentiment analysis task

  sentimentAnalysisModel = await transformersLoad('sentiment-analysis', 'Xenova/distilbert-base-uncased-finetuned-sst-2-english');

}

​

function setup() {

  createCanvas(400, 400);

  txt = createInput();

  button = createButton("Go");

  button.mousePressed(startPredicting);

}

​

function startPredicting() {

  transformersPredict(sentimentAnalysisModel, txt.value(), donePredicting);

  txt.value('');

}

​

function draw() {

  if (label == "POSITIVE") {

    background(lerpColor(color("white"), color("green"), score));

  } else if (label == "NEGATIVE") {

    background(lerpColor(color("white"), color("red"), score));

  }

}

​

function donePredicting(results) {

  console.log(results);

  label = results[0].label;

  score = results[0].score;

}

​