slszeto commited on
Commit
7e76d3c
·
verified ·
1 Parent(s): e0258b8

Update Dockerfile

Browse files
Files changed (1) hide show
  1. Dockerfile +8 -0
Dockerfile CHANGED
@@ -2,12 +2,20 @@ FROM python:3.13.5-slim
2
 
3
  WORKDIR /app
4
 
 
 
 
5
  RUN apt-get update && apt-get install -y \
6
  build-essential \
7
  curl \
8
  git \
9
  && rm -rf /var/lib/apt/lists/*
10
 
 
 
 
 
 
11
  COPY requirements.txt ./
12
  COPY src/ ./src/
13
 
 
2
 
3
  WORKDIR /app
4
 
5
+ RUN mkdir -p /data/.cache/huggingface/transformers \
6
+ && mkdir -p /data/.cache/sentence_transformers
7
+
8
  RUN apt-get update && apt-get install -y \
9
  build-essential \
10
  curl \
11
  git \
12
  && rm -rf /var/lib/apt/lists/*
13
 
14
+ ENV HF_HOME=/data/.cache/huggingface
15
+ ENV TRANSFORMERS_CACHE=/data/.cache/huggingface/transformers
16
+ ENV SENTENCE_TRANSFORMERS_HOME=/data/.cache/sentence_transformers
17
+ ENV HF_HUB_ENABLE_HF_TRANSFER=1
18
+
19
  COPY requirements.txt ./
20
  COPY src/ ./src/
21