Yermek68 commited on
Commit
9a51e18
·
verified ·
1 Parent(s): 0f6d525

Update Dockerfile

Browse files
Files changed (1) hide show
  1. Dockerfile +3 -3
Dockerfile CHANGED
@@ -15,10 +15,10 @@ COPY . /app
15
 
16
  # =============== INSTALL DEPENDENCIES ======================
17
  RUN pip install --no-cache-dir --upgrade pip \
 
18
  && pip install --no-cache-dir -r requirements.txt \
19
- && pip install --no-cache-dir "huggingface-hub==0.23.4" \
20
- && pip uninstall -y gradio gradio-client spaces || true
21
-
22
  # =============== SET ENVIRONMENT ======================
23
  ENV HF_HOME=/tmp/huggingface
24
  ENV TRANSFORMERS_CACHE=/tmp/transformers_cache
 
15
 
16
  # =============== INSTALL DEPENDENCIES ======================
17
  RUN pip install --no-cache-dir --upgrade pip \
18
+ && pip uninstall -y gradio gradio-client spaces huggingface-hub || true \
19
  && pip install --no-cache-dir -r requirements.txt \
20
+ && pip install --no-cache-dir "huggingface-hub==0.23.4" "transformers==4.46.0" "torch==2.4.1" \
21
+ && pip check
 
22
  # =============== SET ENVIRONMENT ======================
23
  ENV HF_HOME=/tmp/huggingface
24
  ENV TRANSFORMERS_CACHE=/tmp/transformers_cache