Update Dockerfile

Dockerfile

@@ -1,11 +1,21 @@
-FROM ghcr.io/arnidan/nsfw-api:latest
+FROM node:lts-slim
 
-ENV PORT=7860 
+RUN apt-get update && apt-get install -y python3 build-essential \
+    && rm -rf /var/lib/apt/lists/*
 
-WORKDIR /app
+COPY yarn.lock package.json ./
 
-USER node
+RUN npm_config_build_from_source=true yarn install --prod
 
-EXPOSE $PORT
+COPY src ./src
+COPY tsconfig.json ./
+
+ARG modelType=default
+
+COPY ./models/$modelType ./model
+
+RUN yarn build
+
+EXPOSE 7860
 
 ENTRYPOINT ["yarn", "start"]