diff options
Diffstat (limited to 'src/server/index.ts')
-rw-r--r-- | src/server/index.ts | 6 |
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); }); |