Merge pull request #11 from Neo-Zenith/wy-dev

Wy dev

express/.gitignore

@@ -1 +1,2 @@
-node_modules
+node_modules
+.env

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"

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: {
@@ -21,10 +22,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: process.env.HUGGING_FACE_TOKEN },
+			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");