Update Dockerfile

Dockerfile

@@ -1,4 +1,4 @@
-FROM nvidia/cuda:11.0.3-base-ubuntu20.04
+FROM ubuntu:20.04
 
 ENV MODEL_NAME="llava-1.6-mistral-7b-gguf"
 ENV DEFAULT_MODEL_FILE="llava-v1.6-mistral-7b.Q3_K_XS.gguf"
@@ -7,7 +7,6 @@ ENV DEFAULT_MODEL_BRANCH="main"
 ENV DEFAULT_CLIP_MODEL_FILE="mmproj-model-f16.gguf"
 ENV MODEL_URL="https://huggingface.co/${MODEL_USER}/${MODEL_NAME}/resolve/${DEFAULT_MODEL_BRANCH}/${DEFAULT_MODEL_FILE}"
 ENV CLIP_MODEL_URL="https://huggingface.co/${MODEL_USER}/${MODEL_NAME}/resolve/${DEFAULT_MODEL_BRANCH}/${DEFAULT_CLIP_MODEL_FILE}"
-ENV PATH="/usr/local/cuda/bin:$PATH"
 
 RUN apt update && \
     apt install --no-install-recommends -y build-essential python3 python3-pip wget curl git && \
@@ -15,39 +14,31 @@ RUN apt update && \
 
 WORKDIR /app
 
-RUN apt-get install -y wget && \
-    wget -qO- "https://cmake.org/files/v3.18/cmake-3.18.0-Linux-x86_64.tar.gz" | tar --strip-components=1 -xz -C /usr/local
-
 COPY requirements.txt ./
 
-RUN pip install --upgrade pip && \
+RUN python3 -m pip install --upgrade pip && \
     pip install -r requirements.txt
 
-# Moving back to /app directory
-WORKDIR /app
-
+# Downloading the models
 RUN echo ${MODEL_URL} && \
-    wget -O /app/${DEFAULT_MODEL_FILE} ${MODEL_URL}
-
-RUN echo ${CLIP_MODEL_URL} && \
+    wget -O /app/${DEFAULT_MODEL_FILE} ${MODEL_URL} && \
+    echo ${CLIP_MODEL_URL} && \
     wget -O /app/${DEFAULT_CLIP_MODEL_FILE} ${CLIP_MODEL_URL}
-    
-RUN useradd -m -u 1000 user
 
-RUN mkdir -p /home/user/app && \
+# Creating a non-root user
+RUN useradd -m -u 1000 user && \
+    mkdir -p /home/user/app && \
     mv /app/${DEFAULT_MODEL_FILE} /home/user/app && \
-    mv /app/${DEFAULT_CLIP_MODEL_FILE} /home/user/app
-
-RUN chown -R user:user /home/user/app
+    mv /app/${DEFAULT_CLIP_MODEL_FILE} /home/user/app && \
+    chown -R user:user /home/user/app
 
 USER user
 ENV HOME=/home/user \
-	PATH=/home/user/.local/bin:$PATH \
-    MODEL_NAME=${MODEL_NAME} \
-    MODEL_FILE=/home/user/app/${DEFAULT_MODEL_FILE} 
+    PATH=/home/user/.local/bin:$PATH
 
 WORKDIR $HOME/app
 
+# Copying the rest of your application
 COPY --chown=user . .
 
 RUN ls -al