Update Dockerfile

Dockerfile

@@ -1,12 +1,39 @@
 FROM ghcr.io/huggingface/chat-ui:latest AS base
 
-FROM ghcr.io/huggingface/text-generation-inference:latest AS final
+# Use a CPU-specific base image for text-generation-inference
+FROM ubuntu:22.04 AS final
 
 ARG MODEL_NAME
 ENV MODEL_NAME=${MODEL_NAME}
 
 ENV TZ=Europe/Paris \
-    PORT=3000
+    PORT=3000 \
+    HUGGING_FACE_HUB_TOKEN="" \
+    MONGO_URL="mongodb://localhost:27017" \
+    DB_NAME="chat-ui"
+
+# Install system dependencies for a CPU environment
+RUN apt-get update && \
+    apt-get install -y --no-install-recommends \
+    wget \
+    curl \
+    git \
+    build-essential \
+    ca-certificates \
+    libssl-dev \
+    libopenblas-dev \
+    libopenblas-base \
+    libatlas-base-dev \
+    libgl1-mesa-glx \
+    libglib2.0-0 \
+    python3 \
+    python3-pip \
+    && rm -rf /var/lib/apt/lists/*
+
+# Install text-generation-inference from source for CPU optimization
+RUN git clone https://github.com/huggingface/text-generation-inference.git /app/text-generation-inference && \
+    cd /app/text-generation-inference && \
+    pip3 install --no-cache-dir .
 
 # mongo installation
 RUN curl -fsSL https://www.mongodb.org/static/pgp/server-7.0.asc | \
@@ -43,7 +70,6 @@ ENV HOME=/home/user \
 RUN npm config set prefix /home/user/.local
 RUN npm install -g dotenv-cli
 
-
 # copy chat-ui from base image
 COPY --from=base --chown=1000 /app/node_modules /app/node_modules
 COPY --from=base --chown=1000 /app/package.json /app/package.json