Update Dockerfile to include notebook requirement files

Dockerfile

@@ -72,6 +72,10 @@ RUN apt-get update && \
 COPY check_gpu.sh /root/check_gpu.sh
 RUN chmod +x /root/check_gpu.sh
 
+# Copy notebook requirement files
+COPY notebooks/requirements_cpu.txt /home/user/notebooks/requirements_cpu.txt
+COPY notebooks/requirements_gpu.txt /home/user/notebooks/requirements_gpu.txt
+
 # Conditional installation based on GPU availability
 RUN if /root/check_gpu.sh; then \
         echo "GPU detected. Installing GPU-specific Python packages." && \
@@ -81,9 +85,6 @@ RUN if /root/check_gpu.sh; then \
         cp /home/user/notebooks/requirements_cpu.txt /home/user/requirements_specific.txt; \
     fi
 
-# Copy notebooks directory to a temporary location
-COPY --chown=user notebooks /home/user/notebooks
-
 #######################################
 # End root user section
 #######################################
@@ -92,10 +93,6 @@ USER user
 
 # Python packages
 COPY base_requirements.txt /home/user/base_requirements.txt
-COPY notebooks/requirements.txt /home/user/notebooks/requirements.txt
-COPY notebooks/requirements_cpu.txt /home/user/notebooks/requirements_cpu.txt
-COPY notebooks/requirements_gpu.txt /home/user/notebooks/requirements_gpu.txt
-
 RUN pip install --no-cache-dir --upgrade -r /home/user/base_requirements.txt -r /home/user/requirements_specific.txt
 
 # Copy the current directory contents into the container at $HOME/app setting the owner to the user