JulianPhillips commited on
Commit
abbd5e7
·
verified ·
1 Parent(s): 95211d5

Update Dockerfile

Browse files
Files changed (1) hide show
  1. Dockerfile +2 -1
Dockerfile CHANGED
@@ -17,7 +17,7 @@ RUN apt-get update && apt-get install -y \
17
  && rm -rf /var/lib/apt/lists/*
18
 
19
  # Install Python packages including Hugging Face Transformers, TorchScript, Flask, TensorFlow, TensorFlow Hub, and OpenCV
20
- RUN pip install --no-cache-dir \
21
  torch \
22
  torchvision \
23
  transformers \
@@ -28,6 +28,7 @@ RUN pip install --no-cache-dir \
28
  tensorflow_hub \
29
  opencv-python
30
 
 
31
  # Set Hugging Face cache to a guaranteed writable directory
32
  ENV TRANSFORMERS_CACHE=/app/cache
33
  ENV HF_HOME=/app/cache
 
17
  && rm -rf /var/lib/apt/lists/*
18
 
19
  # Install Python packages including Hugging Face Transformers, TorchScript, Flask, TensorFlow, TensorFlow Hub, and OpenCV
20
+ RUN pip install --no-cache-dir --ignore-installed \
21
  torch \
22
  torchvision \
23
  transformers \
 
28
  tensorflow_hub \
29
  opencv-python
30
 
31
+
32
  # Set Hugging Face cache to a guaranteed writable directory
33
  ENV TRANSFORMERS_CACHE=/app/cache
34
  ENV HF_HOME=/app/cache