Update Dockerfile

Dockerfile

@@ -14,9 +14,6 @@ RUN apt-get update && \
     apt-get install -y git && \
     git clone https://github.com/lydianish/RoLASER.git /app/RoLASER
 
-# Switch to the non-root user
-USER user
-
 # Clone the RoLASER repository into the container
 WORKDIR /app/RoLASER
 
@@ -25,7 +22,6 @@ COPY --chown=user environment.yml /app/RoLASER/environment.yml
 RUN conda env create -f environment.yml
 RUN echo "conda activate rolaser_env" >> ~/.bashrc
 ENV PATH /opt/conda/envs/rolaser_env/bin:$PATH
-ENV ROLASER /app/RoLASER
 
 # Install PyTorch 1.10.1
 RUN pip3 install --no-cache-dir torch==1.10.1 torchvision==0.11.2 torchaudio==0.10.1 -f https://download.pytorch.org/whl/cu102/torch_stable.html
@@ -53,6 +49,10 @@ RUN git checkout rolaser \
 # Install other RoLASER dependencies
 WORKDIR /app/RoLASER 
 RUN pip3 install --no-cache-dir -r /app/ROLASER_requirements.txt
+ENV ROLASER /app/RoLASER
+
+# Switch to the non-root user
+USER user
 
 # Download models
 WORKDIR /app/RoLASER/models