Create inference.js

KOSASIH · Nov 20, 2024 · b312895 · b312895
1 parent e90dcd3
commit b312895
Showing 1 changed file with 5 additions and 0 deletions.
diff --git a/dapps-builder/src/components/AIEngine/inference.js b/dapps-builder/src/components/AIEngine/inference.js
@@ -0,0 +1,5 @@
+export const makePrediction = async (model, input) => {
+    const inputTensor = tf.tensor2d([input]);
+    const prediction = model.predict(inputTensor);
+    return prediction.arraySync();
+};