// files. Make sure to duplicate it, rather than copying

// and pasting code :)

​

let replicate_api_proxy = "https://splashy-rambunctious-leader.glitch.me/";

let bg;

let txt = "";

let button;

​

function preload() {

  bg = loadImage("bg.png");

}

​

function setup() {

  createCanvas(978, 550);

​

  console.log("Starting prediction, this might take a bit");

  startPredicting();

}

​

function startPredicting() {

  let modelInput = {

    prompt: "Well hello friends",

    max_new_tokens: 128,

    temperature: 0.75,

    top_p: 0.9,

    top_k: 50,

  };

​

  predictReplicate(

    // replace this by the model string copied in the previous step

    "itpnyu/llama2-gohai-ak-videos1:c323619b74ff4e13b6c211783e7e4f15415eb51d92bc571cc7db655e361bee41",

    // compare with the foundation model:

    //"meta/llama-2-7b:73001d654114dad81ec65da3b834e2f691af1e1526453189b7bf36fb3f32d0f9",

    modelInput,

    donePredicting

  );

}

​

function donePredicting(results, input) {

  console.log(results);

  txt = input.prompt + results.join("");

​

  if (!button) {

    button = createButton("again");

  }

  button.position(40, height - 60);

  button.mousePressed(startPredicting);

}

​

function draw() {

  image(bg, 0, 0, width, height);

​

  if (txt.length > 0) {

    fill(255, 150);

    noStroke();

    rect(40, 40, width - 80, height - 100);

    fill(0);

    textSize(26);

    textLeading(38);

    text(txt, 50, 50, width - 100, width - 120);

  }

}

​