Fix name and API

Author: pearmini

examples/imageClassifier-transformer-single-image/sketch.js

@@ -39,10 +39,7 @@ function gotResult(results) {
   stroke(0);
   textSize(18);
   label = "Label: " + results[0].label;
-  // Handle both 'score' (transformer) and 'confidence' (other models) formats
-  const score =
-    results[0].score !== undefined ? results[0].score : results[0].confidence;
-  confidence = "Confidence: " + nf(score, 0, 2);
+  confidence = "Confidence: " + nf(results[0].score, 0, 2);
   text(label, 10, 360);
   text(confidence, 10, 380);
 }

src/ImageClassifier/index.js

@@ -18,7 +18,7 @@ import callCallback from "../utils/callcallback";
 import { imgToTensor, mediaReady } from "../utils/imageUtilities";
 import handleOptions from "../utils/handleOptions";
 import { handleModelName } from "../utils/handleOptions";
-import { Transformer } from "./transformer";
+import { ImageClassifierTransformer } from "./transformer";
 
 const IMAGE_SIZE = 224;
 const MODEL_OPTIONS = ["mobilenet", "darknet", "darknet-tiny", "doodlenet"];
@@ -318,7 +318,7 @@ const imageClassifier = (modelName, optionsOrCallback, cb) => {
   const { string, options = {}, callback } = args;
   const instance =
     string === "transformer"
-      ? new Transformer(options, callback)
+      ? new ImageClassifierTransformer(options, callback)
       : new ImageClassifier(string, options, callback);
   return instance;
 };

src/ImageClassifier/transformer.js

@@ -1,6 +1,6 @@
 import { pipeline } from "@huggingface/transformers";
 
-export class Transformer {
+export class ImageClassifierTransformer {
   constructor(options, callback) {
     this.classifier = null;
     this.needToStop = false;