eggacheb commited on
Commit
fe558b7
·
verified ·
1 Parent(s): f2491c5

Update Dockerfile

Browse files
Files changed (1) hide show
  1. Dockerfile +3 -3
Dockerfile CHANGED
@@ -5,7 +5,8 @@ FROM pytorch/pytorch:${TORCH_VERSION}-cuda${CUDA_VERSION}-cudnn8-runtime
5
  WORKDIR /opt/NekoImageGallery
6
 
7
  COPY requirements.txt .
8
-
 
9
  RUN PYTHONDONTWRITEBYTECODE=1 pip install --no-cache-dir -r requirements.txt
10
 
11
  RUN mkdir -p /opt/models && \
@@ -16,8 +17,7 @@ RUN mkdir -p /opt/models && \
16
  'PaddleOCR2Pytorch/ch_ptocr_mobile_v2.0_cls_infer.pth' 'PaddleOCR2Pytorch/configs/det/ch_PP-OCRv4/ch_PP-OCRv4_det_student.yml' \
17
  'PaddleOCR2Pytorch/configs/rec/PP-OCRv4/ch_PP-OCRv4_rec.yml' 'ppocr_keys_v1.txt' --local-dir /opt/models/ocr && \
18
  rm -rf /root/.cache/huggingface
19
- RUN mkdir /root/.cache/huggingface/hub
20
- RUN chmod -R 777 /root/.cache/huggingface/hub
21
 
22
  ENV APP_MODEL__CLIP=/opt/models/clip
23
  ENV APP_MODEL__BERT=/opt/models/bert
 
5
  WORKDIR /opt/NekoImageGallery
6
 
7
  COPY requirements.txt .
8
+ RUN mkdir /root/.cache/huggingface/hub
9
+ RUN chmod -R 777 /root/.cache/huggingface/hub
10
  RUN PYTHONDONTWRITEBYTECODE=1 pip install --no-cache-dir -r requirements.txt
11
 
12
  RUN mkdir -p /opt/models && \
 
17
  'PaddleOCR2Pytorch/ch_ptocr_mobile_v2.0_cls_infer.pth' 'PaddleOCR2Pytorch/configs/det/ch_PP-OCRv4/ch_PP-OCRv4_det_student.yml' \
18
  'PaddleOCR2Pytorch/configs/rec/PP-OCRv4/ch_PP-OCRv4_rec.yml' 'ppocr_keys_v1.txt' --local-dir /opt/models/ocr && \
19
  rm -rf /root/.cache/huggingface
20
+
 
21
 
22
  ENV APP_MODEL__CLIP=/opt/models/clip
23
  ENV APP_MODEL__BERT=/opt/models/bert