diff --git a/src/server.js b/src/server.js index 33790a4..b4bd70f 100644 --- a/src/server.js +++ b/src/server.js @@ -1,13 +1,19 @@ -// Import dotenv file if not in production -if (process.env.NODE_ENV !== "production") { - require("dotenv").config(); -} - const {env, pipeline} = require("@xenova/transformers"); const path = require("path"); env.allowRemoteModels = false; -env.localModelPath = path.resolve(__dirname, "./models/transformers").replace(/\\/g, "/"); env.backends.onnx.wasm.proxy = false; +env.backends.onnx.wasm.numThreads = 1; + +// Import dotenv file if not in production +if (process.env.NODE_ENV !== "production") { + require("dotenv").config(); + env.localModelPath = path.resolve(__dirname, "./models/transformers").replace(/\\/g, "/"); +} else { + env.localModelPath = "/app/src/models/transformers"; +} +console.log(process.platform, process.arch, process.version); +console.log(env.localModelPath); + const models = require("./models"); const db = models.db;