Update Dockerfile

Dockerfile

@@ -3,6 +3,15 @@ FROM  nvidia/cuda:12.4.1-runtime-ubuntu22.04
 
 # Set the working directory
 WORKDIR /DocQA
+# Install dependencies
+RUN apt-get update && apt-get install -y \
+    git \
+    python3.10 \
+    python3-pip \
+    && apt-get clean
+RUN apt-get install poppler-utils -y
+
+RUN pip3 install torch --index-url https://download.pytorch.org/whl/cu121
 RUN useradd -m -u 1000 user
 
 USER user
@@ -14,15 +23,7 @@ WORKDIR $HOME/app
 # Copy the requirements.txt file
 COPY --chown=user requirements.txt $HOME/app
 
-# Install dependencies
-RUN apt-get update && apt-get install -y \
-    git \
-    python3.10 \
-    python3-pip \
-    && apt-get clean
-RUN apt-get install poppler-utils -y
 RUN pip install --no-cache-dir -r requirements.txt 
-RUN pip3 install torch --index-url https://download.pytorch.org/whl/cu121
 
 # Copy the rest of the application codeDocQA
 COPY --chown=user images $HOME/app