From 694a1515705439c481eeff9541ae5c3c219f5f2b Mon Sep 17 00:00:00 2001 From: Woon Yee Ng Date: Sat, 20 Jan 2024 19:31:40 +0800 Subject: [PATCH 1/2] add xlnet model --- express/package-lock.json | 3 +-- express/server.js | 31 +++++++++++++++++++++++++++++++ 2 files changed, 32 insertions(+), 2 deletions(-) diff --git a/express/package-lock.json b/express/package-lock.json index 9066436..f69b9c6 100644 --- a/express/package-lock.json +++ b/express/package-lock.json @@ -13,8 +13,7 @@ "express": "^4.18.2", "nodemon": "^3.0.3", "socket.io": "^4.7.4" - }, - "devDependencies": {} + } }, "node_modules/@socket.io/component-emitter": { "version": "3.1.0", diff --git a/express/server.js b/express/server.js index d540eed..b57f05b 100644 --- a/express/server.js +++ b/express/server.js @@ -21,10 +21,41 @@ app.use( ); app.use(express.json()); +async function queryXLNET(data) { + const response = await fetch( + "https://api-inference.huggingface.co/models/xlnet/xlnet-base-cased", + { + headers: { Authorization: "Bearer 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" }, + method: "POST", + body: JSON.stringify(data), + } + ); + const result = await response.json(); + return result; +} + + + app.get("/", (req, res) => { return res.json("hi"); }); +app.get("/xlnet", async (req, res) => { + try { + const response = await queryXLNET({ + "inputs": "Can you please let us know more details about your " + }); + var ans = response[0]?.generated_text || ''; + return res.json(ans); + } catch (error) { + console.error(error); + return res.status(500).json({ error: "Internal Server Error" }); + } +}); + + + + app.get("/output", (req, res) => { io.emit("new-output", "hello"); return res.json("successs"); From 81e18da9fb8ef484ac477898c30179bed07fbe19 Mon Sep 17 00:00:00 2001 From: Woon Yee Ng Date: Sat, 20 Jan 2024 19:40:16 +0800 Subject: [PATCH 2/2] resolve gitguardian check --- express/.gitignore | 3 ++- express/package-lock.json | 17 +++++++++++++++++ express/package.json | 1 + express/server.js | 3 ++- 4 files changed, 22 insertions(+), 2 deletions(-) diff --git a/express/.gitignore b/express/.gitignore index b512c09..1dcef2d 100644 --- a/express/.gitignore +++ b/express/.gitignore @@ -1 +1,2 @@ -node_modules \ No newline at end of file +node_modules +.env \ No newline at end of file diff --git a/express/package-lock.json b/express/package-lock.json index f69b9c6..320428e 100644 --- a/express/package-lock.json +++ b/express/package-lock.json @@ -10,6 +10,7 @@ "license": "ISC", "dependencies": { "cors": "^2.8.5", + "dotenv": "^16.3.2", "express": "^4.18.2", "nodemon": "^3.0.3", "socket.io": "^4.7.4" @@ -273,6 +274,17 @@ "npm": "1.2.8000 || >= 1.4.16" } }, + "node_modules/dotenv": { + "version": "16.3.2", + "resolved": "https://registry.npmjs.org/dotenv/-/dotenv-16.3.2.tgz", + "integrity": "sha512-HTlk5nmhkm8F6JcdXvHIzaorzCoziNQT9mGxLPVXW8wJF1TiGSL60ZGB4gHWabHOaMmWmhvk2/lPHfnBiT78AQ==", + "engines": { + "node": ">=12" + }, + "funding": { + "url": "https://github.com/motdotla/dotenv?sponsor=1" + } + }, "node_modules/ee-first": { "version": "1.1.1", "resolved": "https://registry.npmjs.org/ee-first/-/ee-first-1.1.1.tgz", @@ -1428,6 +1440,11 @@ "resolved": "https://registry.npmjs.org/destroy/-/destroy-1.2.0.tgz", "integrity": "sha512-2sJGJTaXIIaR1w4iJSNoN0hnMY7Gpc/n8D4qSCJw8QqFWXf7cuAgnEHxBpweaVcPevC2l3KpjYCx3NypQQgaJg==" }, + "dotenv": { + "version": "16.3.2", + "resolved": "https://registry.npmjs.org/dotenv/-/dotenv-16.3.2.tgz", + "integrity": "sha512-HTlk5nmhkm8F6JcdXvHIzaorzCoziNQT9mGxLPVXW8wJF1TiGSL60ZGB4gHWabHOaMmWmhvk2/lPHfnBiT78AQ==" + }, "ee-first": { "version": "1.1.1", "resolved": "https://registry.npmjs.org/ee-first/-/ee-first-1.1.1.tgz", diff --git a/express/package.json b/express/package.json index 1f0b27f..4f73ee1 100644 --- a/express/package.json +++ b/express/package.json @@ -11,6 +11,7 @@ "license": "ISC", "dependencies": { "cors": "^2.8.5", + "dotenv": "^16.3.2", "express": "^4.18.2", "nodemon": "^3.0.3", "socket.io": "^4.7.4" diff --git a/express/server.js b/express/server.js index b57f05b..e745f61 100644 --- a/express/server.js +++ b/express/server.js @@ -5,6 +5,7 @@ const server = require("http").createServer(app); // const helmet = require("helmet"); const cors = require("cors"); // const authRouters = require("./routers/authRouter"); +require('dotenv').config() const io = new Server(server, { cors: { @@ -25,7 +26,7 @@ async function queryXLNET(data) { const response = await fetch( "https://api-inference.huggingface.co/models/xlnet/xlnet-base-cased", { - headers: { Authorization: "Bearer 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" }, + headers: { Authorization: process.env.HUGGING_FACE_TOKEN }, method: "POST", body: JSON.stringify(data), }