tensorflow model, server posting fix needed

1 files changed, 5 insertions, 3 deletions

diff --git a/src/server/index.ts b/src/server/index.ts
index e7d49579d..ac803a253 100644
--- a/src/server/index.ts
+++ b/src/server/index.ts
@@ -688,13 +688,15 @@ app.use(RouteStore.corsProxy, (req, res) => {
 
 let recommender = new Recommender();
 recommender.testModel();
-recommender.trainModel();
-recommender.arxivRequest("Triangle-GAN");
 
 app.post("/recommender", async (req, res) => {
     let keyphrases = req.body.keyphrases;
     let wordvecs = await recommender.vectorize(keyphrases);
-    res.send(wordvecs);
+    let embedding: number[][] = [];
+    if (wordvecs && wordvecs.array()) {
+        wordvecs.array().then(array => embedding = array as number[][]);
+    }
+    res.send(embedding);
 });