Update Dockerfile

Dockerfile

@@ -10,12 +10,17 @@ RUN apt-get update && apt-get install --no-install-recommends -y \
   python3.11-venv \
   && apt-get clean && rm -rf /var/lib/apt/lists/*
 
+WORKDIR /app
+
 RUN python3.11 -m venv /opt/venv
 ENV PATH=/opt/venv/bin:$PATH \
     NVIDIA_VISIBLE_DEVICES=all \
     NVIDIA_DRIVER_CAPABILITIES=compute,utility
 
-RUN CMAKE_ARGS="-DLLAMA_CUDA=on" pip3 install --no-cache-dir llama-cpp-python wikipedia
+COPY ./requirements.txt /app/requirements.txt
+RUN pip3 install --no-cache-dir --upgrade -r /app/requirements.txt
+
+RUN CMAKE_ARGS="-DLLAMA_CUDA=on" pip3 install --no-cache-dir llama-cpp-python
 
 FROM qdrant/qdrant:latest
 
@@ -41,25 +46,20 @@ WORKDIR /app
 
 RUN python3 -m venv /opt/venv
 
-ENV PATH=/opt/venv/bin:$PATH \
-    PYTHONUNBUFFERED=1 \
-    NVIDIA_VISIBLE_DEVICES=all \
-    NVIDIA_DRIVER_CAPABILITIES=compute,utility
-
-COPY --from=builder /opt/venv /opt/venv
-
-COPY ./requirements.txt /app/requirements.txt
-RUN pip3 install --no-cache-dir --upgrade -r /app/requirements.txt
-
 RUN useradd -m -u 1000 user
 USER user
 
 ENV HOME=/home/user \
-	PATH=/home/user/.local/bin:$PATH \
+	PATH=/home/user/.local/bin:/opt/venv/bin:$PATH \
     PYTHONPATH=$HOME/app \
+    PYTHONUNBUFFERED=1 \
+    NVIDIA_VISIBLE_DEVICES=all \
+    NVIDIA_DRIVER_CAPABILITIES=compute,utility
 	SYSTEM=spaces \
     HF_HOME=/data/.huggingface 
 
+COPY --from=builder /opt/venv /opt/venv
+
 WORKDIR $HOME/app
 
 COPY --chown=user . $HOME/app