aboutsummaryrefslogtreecommitdiff
path: root/src/server/index.ts
diff options
context:
space:
mode:
Diffstat (limited to 'src/server/index.ts')
-rw-r--r--src/server/index.ts6
1 files changed, 3 insertions, 3 deletions
diff --git a/src/server/index.ts b/src/server/index.ts
index ac803a253..49957775d 100644
--- a/src/server/index.ts
+++ b/src/server/index.ts
@@ -692,9 +692,9 @@ recommender.testModel();
app.post("/recommender", async (req, res) => {
let keyphrases = req.body.keyphrases;
let wordvecs = await recommender.vectorize(keyphrases);
- let embedding: number[][] = [];
- if (wordvecs && wordvecs.array()) {
- wordvecs.array().then(array => embedding = array as number[][]);
+ let embedding: Float32Array = new Float32Array();
+ if (wordvecs && wordvecs.dataSync()) {
+ embedding = wordvecs.dataSync() as Float32Array;
}
res.send(embedding);
});