rm extends video since we feed in out inputs - note this could be interesting for future video support

Author: joeyklee

src/UNET/index.js

@@ -10,13 +10,12 @@ Image Classifier using pre-trained networks
 import * as tf from '@tensorflow/tfjs';
 import callCallback from '../utils/callcallback';
 import { array3DToImage } from '../utils/imageUtilities';
-import Video from '../utils/Video';
 import p5Utils from '../utils/p5Utils';
 
 const URL = 'https://raw.githubusercontent.com/zaidalyafeai/HostedModels/master/unet-128/model.json';
 const imageSize = 128;
 
-class UNET extends Video {
+class UNET {
   /**
    * Create UNET class. 
    * @param {HTMLVideoElement | HTMLImageElement} video - The video or image to be used for segmentation.
@@ -25,16 +24,12 @@ class UNET extends Video {
    *    that will be resolved once the model has loaded.
    */
   constructor(video, options, callback) {
-    super(video, imageSize);
     this.modelReady = false;
     this.isPredicting = false;
     this.ready = callCallback(this.loadModel(), callback);
   }
 
   async loadModel() {
-    if (this.videoElt && !this.video) {
-      this.video = await this.loadVideo();
-    }
     this.model = await tf.loadLayersModel(URL);
     this.modelReady = true;
     return this;