no nesting

mitmedialab · Jun 25, 2024 · f9930be · f9930be
1 parent f835ac8
commit f9930be
Showing 1 changed file with 3 additions and 5 deletions.
diff --git a/extensions/src/teachableMachine/index.ts b/extensions/src/teachableMachine/index.ts
@@ -147,11 +147,9 @@ export default class teachableMachine extends extension({
     const { model, modelType } = this.predictionState[modelUrl];
     switch (modelType) {
       case this.ModelType.IMAGE:
-        if (frame) {
-          const imageBitmap = await createImageBitmap(frame);
-          return await model.predict(imageBitmap);
-        }
-        return null;
+        if (!frame) return null;
+        const imageBitmap = await createImageBitmap(frame);
+        return await model.predict(imageBitmap);
       case this.ModelType.POSE:
         const { pose, posenetOutput } = await model.estimatePose(frame);
         return await model.predict(posenetOutput);