CyberCoder225 commited on
Commit
46d94c0
·
verified ·
1 Parent(s): a58d40c

Update Dockerfile

Browse files
Files changed (1) hide show
  1. Dockerfile +17 -2
Dockerfile CHANGED
@@ -1,9 +1,24 @@
1
- FROM python:3.9
 
 
 
 
 
 
 
 
2
  RUN useradd -m -u 1000 user
3
  USER user
4
  ENV HOME=/home/user \
5
  PATH=/home/user/.local/bin:$PATH
6
  WORKDIR $HOME/app
 
 
 
 
 
 
 
7
  COPY --chown=user . $HOME/app
8
- RUN pip install --no-cache-dir flask llama-cpp-python huggingface-hub
9
  CMD ["python", "app.py"]
 
1
+ FROM python:3.10-slim
2
+
3
+ # Install system dependencies needed for the brain
4
+ RUN apt-get update && apt-get install -y \
5
+ build-essential \
6
+ cmake \
7
+ python3-dev \
8
+ && rm -rf /var/lib/apt/lists/*
9
+
10
  RUN useradd -m -u 1000 user
11
  USER user
12
  ENV HOME=/home/user \
13
  PATH=/home/user/.local/bin:$PATH
14
  WORKDIR $HOME/app
15
+
16
+ # MAGIC LINE: This installs the pre-compiled version for CPU
17
+ RUN pip install --no-cache-dir \
18
+ huggingface-hub \
19
+ flask \
20
+ llama-cpp-python --extra-index-url https://abetlen.github.io/llama-cpp-python/whl/cpu
21
+
22
  COPY --chown=user . $HOME/app
23
+
24
  CMD ["python", "app.py"]