clearer

TomasGonzalez · Apr 9, 2024 · 45e1baf · 45e1baf
1 parent 890dadc
commit 45e1baf
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/src/components/hands-capture/hooks/useKeyPointClassifier.ts b/src/components/hands-capture/hooks/useKeyPointClassifier.ts
@@ -1,6 +1,8 @@
 import { useEffect, useRef } from 'react';
 import { Landmark, Results } from '@mediapipe/hands';
+
 import * as tf from '@tensorflow/tfjs';
+
 import _ from 'lodash';
 
 const calcLandmarkList = (image, landmarks) => {
@@ -51,13 +53,11 @@ function useKeyPointClassifier() {
   const model = useRef<any>();
 
   const keyPointClassifier = async (landmarkList) => {
-    const result = await model.current
+    return await model.current
       .execute(tf.tensor2d([landmarkList]))
       .squeeze()
       .argMax()
       .data();
-
-    return result;
   };
 
   const processLandmark = async (handLandmarks: Results, image) => {