Spilum.Net/mtf-nlp
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Try a smaller node image

Browse Source
This commit is contained in:
Apher 2026-07-07 23:06:31 +00:00
parent 1142f11724
commit 29f2c0d869
2 changed files with 2 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
Dockerfile
View File

@ -1,4 +1,4 @@
FROM node:20-alpine FROM node:20-bookworm-slim
WORKDIR /app WORKDIR /app

1
src/server.js
View File

@ -66,6 +66,7 @@ function cosineSimilarity(a, b) {
async function initModel() { async function initModel() {
extractor = await pipeline("feature-extraction", "Xenova/all-MiniLM-L6-v2"); extractor = await pipeline("feature-extraction", "Xenova/all-MiniLM-L6-v2");
console.log("MiniLM model loaded");
} }
async function embed(text) { async function embed(text) {