tejani commited on
Commit
91cca98
·
verified ·
1 Parent(s): b7656e5

Update Dockerfile

Browse files
Files changed (1) hide show
  1. Dockerfile +5 -2
Dockerfile CHANGED
@@ -6,6 +6,7 @@ COPY ./requirements.txt /code/requirements.txt
6
 
7
  RUN pip install --no-cache-dir --upgrade -r /code/requirements.txt
8
 
 
9
  RUN apt-get update && apt-get install -y \
10
  libgl1-mesa-glx \
11
  libglib2.0-0 \
@@ -13,16 +14,18 @@ RUN apt-get update && apt-get install -y \
13
  libxrender1 \
14
  && rm -rf /var/lib/apt/lists/*
15
 
 
 
 
16
  RUN useradd -m -u 1000 user
17
 
18
  USER user
19
 
20
  ENV HOME=/home/user \
21
- PATH=/home/user/.local/bin:$PATH
22
 
23
  WORKDIR $HOME/app
24
 
25
  COPY --chown=user . $HOME/app
26
 
27
- # Run app.py directly with Python
28
  CMD ["python", "app.py"]
 
6
 
7
  RUN pip install --no-cache-dir --upgrade -r /code/requirements.txt
8
 
9
+ # Install libGL and other dependencies for OpenCV
10
  RUN apt-get update && apt-get install -y \
11
  libgl1-mesa-glx \
12
  libglib2.0-0 \
 
14
  libxrender1 \
15
  && rm -rf /var/lib/apt/lists/*
16
 
17
+ # Pre-migrate Transformers cache
18
+ RUN python -c "from transformers.utils import move_cache; move_cache()"
19
+
20
  RUN useradd -m -u 1000 user
21
 
22
  USER user
23
 
24
  ENV HOME=/home/user \
25
+ PATH=/home/user/.local/bin:$PATH
26
 
27
  WORKDIR $HOME/app
28
 
29
  COPY --chown=user . $HOME/app
30
 
 
31
  CMD ["python", "app.py"]