Update Dockerfile

Dockerfile

@@ -1,36 +1,21 @@
-FROM nvidia/cuda:10.1-cudnn7-runtime
+FROM python:latest
 
-# Set the working directory
-WORKDIR $HOME/app
+ENV PYTHONUNBUFFERED 1
 
-# Copy the requirements file
-COPY requirements.txt .
-
-RUN apt-get update && \
-    apt-get install -y --no-install-recommends \
-        build-essential \
-        python3-dev \
-        python3-pip \
-        libglib2.0-0 && \
-    rm -rf /var/lib/apt/lists/*
+WORKDIR /app
 
+RUN export PATH="/usr/local/cuda/bin:$PATH"
+RUN wget -qO- "https://cmake.org/files/v3.17/cmake-3.17.0-Linux-x86_64.tar.gz" | tar --strip-components=1 -xz -C /usr/local
 RUN CMAKE_ARGS="-DLLAMA_CUBLAS=on" FORCE_CMAKE=1 pip install llama-cpp-python --no-cache-dir
-
+COPY requirements.txt ./
 RUN pip install --upgrade pip && \
-    pip install --no-cache-dir --upgrade -r requirements.txt
-
-# Copy the installed packages to a new image
-FROM nvidia/cuda:10.1-cudnn7-runtime
-COPY --from=0 /root/.local /root/.local
-ENV PATH=/root/.local/bin:$PATH
+    pip install -r requirements.txt
 
 EXPOSE 7860
 
 RUN useradd -m -u 1000 user
 USER user
 
-# Set the working directory and copy the application code
-WORKDIR /app
 COPY --chown=user . $HOME/app
 
 RUN ls -al