phuc0201 commited on
Commit
3c7e29c
·
verified ·
1 Parent(s): d367460

Update Dockerfile

Browse files
Files changed (1) hide show
  1. Dockerfile +6 -4
Dockerfile CHANGED
@@ -1,13 +1,15 @@
1
  FROM python:3.10-slim
2
 
3
- ENV HF_HOME=/app/.cache
4
-
5
  WORKDIR /app
6
- COPY . /app
 
 
7
 
8
  RUN apt-get update && apt-get install -y git && \
 
 
9
  pip install --no-cache-dir sentence-transformers==2.2.2 transformers==4.30.2 uvicorn fastapi
10
 
11
- RUN mkdir -p /app/.cache
12
 
13
  CMD ["uvicorn", "app:app", "--host", "0.0.0.0", "--port", "7860"]
 
1
  FROM python:3.10-slim
2
 
 
 
3
  WORKDIR /app
4
+
5
+ ENV TRANSFORMERS_CACHE=/tmp/hf_cache
6
+ RUN mkdir -p /tmp/hf_cache
7
 
8
  RUN apt-get update && apt-get install -y git && \
9
+ pip install --no-cache-dir wheel && \
10
+ pip install "huggingface_hub<0.17" && \
11
  pip install --no-cache-dir sentence-transformers==2.2.2 transformers==4.30.2 uvicorn fastapi
12
 
13
+ COPY . /app
14
 
15
  CMD ["uvicorn", "app:app", "--host", "0.0.0.0", "--port", "7860"]