correcting pose detection issue

mitmedialab · Jul 25, 2024 · 3a93d5b · 3a93d5b
1 parent d671d1d
commit 3a93d5b
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/extensions/src/teachableMachine/index.ts b/extensions/src/teachableMachine/index.ts
@@ -151,6 +151,7 @@ export default class teachableMachine extends extension({
         const imageBitmap = await createImageBitmap(frame);
         return await model.predict(imageBitmap);
       case this.ModelType.POSE:
+        if (!frame) return null;
         const { pose, posenetOutput } = await model.estimatePose(frame);
         return await model.predict(posenetOutput);
       case this.ModelType.AUDIO: