mike23415 commited on
Commit
363cac0
·
verified ·
1 Parent(s): eaa1ef7

Update Dockerfile

Browse files
Files changed (1) hide show
  1. Dockerfile +6 -1
Dockerfile CHANGED
@@ -8,6 +8,11 @@ RUN apt-get update && apt-get install -y \
8
  libglib2.0-0 \
9
  && rm -rf /var/lib/apt/lists/*
10
 
 
 
 
 
 
11
  # Copy requirements first to leverage Docker cache
12
  COPY requirements.txt .
13
  RUN pip install --no-cache-dir -r requirements.txt
@@ -16,7 +21,7 @@ RUN pip install --no-cache-dir -r requirements.txt
16
  COPY app.py .
17
 
18
  # Create uploads directory
19
- RUN mkdir -p uploads
20
 
21
  # Expose port
22
  EXPOSE 5000
 
8
  libglib2.0-0 \
9
  && rm -rf /var/lib/apt/lists/*
10
 
11
+ # Create a directory for the Hugging Face cache and make it writable
12
+ RUN mkdir -p /huggingface_cache && chmod 777 /huggingface_cache
13
+ ENV TRANSFORMERS_CACHE=/huggingface_cache
14
+ ENV HF_HOME=/huggingface_cache
15
+
16
  # Copy requirements first to leverage Docker cache
17
  COPY requirements.txt .
18
  RUN pip install --no-cache-dir -r requirements.txt
 
21
  COPY app.py .
22
 
23
  # Create uploads directory
24
+ RUN mkdir -p uploads && chmod 777 uploads
25
 
26
  # Expose port
27
  EXPOSE 5000