shivam98 commited on
Commit
b4b773b
·
verified ·
1 Parent(s): aec53ab

Update Dockerfile

Browse files
Files changed (1) hide show
  1. Dockerfile +4 -5
Dockerfile CHANGED
@@ -2,25 +2,24 @@ FROM pytorch/pytorch:2.1.0-cuda11.8-cudnn8-runtime
2
 
3
  ENV DEBIAN_FRONTEND=noninteractive
4
  ENV TZ=Etc/UTC
5
- ENV HF_HOME=/data/huggingface # <-- use this instead of TRANSFORMERS_CACHE
 
 
6
 
7
  WORKDIR /app
8
 
9
- # Install dependencies
10
  RUN apt-get update && apt-get install -y \
11
  ffmpeg \
12
  git \
13
  tzdata \
14
  && rm -rf /var/lib/apt/lists/*
15
 
16
- # Create Hugging Face cache folder in writable space
17
  RUN mkdir -p /data/huggingface
18
 
19
- # Install Python packages
20
  COPY requirements.txt .
21
  RUN pip install --upgrade pip && pip install -r requirements.txt
22
 
23
- # Copy app code
24
  COPY main.py .
25
 
26
  EXPOSE 7860
 
2
 
3
  ENV DEBIAN_FRONTEND=noninteractive
4
  ENV TZ=Etc/UTC
5
+
6
+ # Use /data as HF_HOME to avoid permission issues
7
+ ENV HF_HOME=/data/huggingface
8
 
9
  WORKDIR /app
10
 
 
11
  RUN apt-get update && apt-get install -y \
12
  ffmpeg \
13
  git \
14
  tzdata \
15
  && rm -rf /var/lib/apt/lists/*
16
 
17
+ # Create cache directory in writable location
18
  RUN mkdir -p /data/huggingface
19
 
 
20
  COPY requirements.txt .
21
  RUN pip install --upgrade pip && pip install -r requirements.txt
22
 
 
23
  COPY main.py .
24
 
25
  EXPOSE 7860