Adedoyinjames commited on
Commit
f5af1f2
·
verified ·
1 Parent(s): c690de7

Update Dockerfile

Browse files
Files changed (1) hide show
  1. Dockerfile +7 -3
Dockerfile CHANGED
@@ -5,8 +5,13 @@ ENV PYTHONUNBUFFERED=1
5
 
6
  WORKDIR /app
7
 
8
- # system deps for some transformers backends (optional; keep minimal)
9
- RUN apt-get update && apt-get install -y git build-essential && rm -rf /var/lib/apt/lists/*
 
 
 
 
 
10
 
11
  COPY requirements.txt .
12
  RUN pip install --no-cache-dir -r requirements.txt
@@ -15,5 +20,4 @@ COPY . .
15
 
16
  EXPOSE 7860
17
 
18
- # default model can be overridden via MODEL_ID env when launching
19
  CMD ["uvicorn", "app:app", "--host", "0.0.0.0", "--port", "7860"]
 
5
 
6
  WORKDIR /app
7
 
8
+ # create writable cache dir and set env vars
9
+ RUN mkdir -p /app/.cache \
10
+ && chown -R root:root /app/.cache
11
+ ENV HF_HOME=/app/.cache
12
+ ENV TRANSFORMERS_CACHE=/app/.cache
13
+ ENV HF_DATASETS_CACHE=/app/.cache
14
+ ENV XDG_CACHE_HOME=/app/.cache
15
 
16
  COPY requirements.txt .
17
  RUN pip install --no-cache-dir -r requirements.txt
 
20
 
21
  EXPOSE 7860
22
 
 
23
  CMD ["uvicorn", "app:app", "--host", "0.0.0.0", "--port", "7860"]