fix: do not crash on empty request (no tensors)

julianpoy · Dec 28, 2023 · fd8560a · fd8560a
1 parent 77282f3
commit fd8560a
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/src/predict.js b/src/predict.js
@@ -26,6 +26,8 @@ const predict = async (sentences) => {
     tensors.push(tensor);
   }
 
+  if (!tensors.length) return [];
+
   const tensor = tf.concat(tensors);
 
   const resultTensors = classifierModel.predict(tensor);