sketch.js

  // Classifier Variable
  let classifier;
  // Model URL
  // let imageModelURL = 'https://teachablemachine.withgoogle.com/models/fpYp1SvwA/';
  let imageModelURL = 'https://teachablemachine.withgoogle.com/models/SjjC-vWdY/';
  
  // Video
  let video;
  let flippedVideo;
  // To store the classification
  let label = "";

  // Load the model first
  function preload() {
    classifier = ml5.imageClassifier(imageModelURL + 'model.json');
  }

  function setup() {
    createCanvas(900, 550);
    // Create the video
    video = createCapture(VIDEO);
    video.size(800, 500);
    video.hide();

    flippedVideo = ml5.flipImage(video);
    // Start classifying
    classifyVideo();
  }

  function draw() {
    background(0);
    // Draw the video
    image(flippedVideo, 0, 0);

    // Draw the label
    fill(255);
    textSize(50);
    textAlign(CENTER);
    text(label, width / 2, height - 4);
  }

  // Get a prediction for the current video frame
  function classifyVideo() {
    flippedVideo = ml5.flipImage(video)
    classifier.classify(flippedVideo, gotResult);
    flippedVideo.remove();

  }

  // When we get a result
  function gotResult(error, results) {

    if (error) {
      console.error(error);
      return;
    }
    
    label = results[0].label;
    // Classifiy again!
    classifyVideo();
  }