Update Dockerfile

Dockerfile

@@ -41,22 +41,29 @@ RUN pip install submodules/simple-knn && \
 #     sed -i '154s/p_view.z <= 0.2f/p_view.z <= 0.001f/' submodules/diff-gaussian-rasterization/cuda_rasterizer/auxiliary.h && \
 #     pip install submodules/diff-gaussian-rasterization
 
+RUN pip install -r requirements.txt
+RUN pip install imageio[ffmpeg]
+
 # Set the working directory
+RUN chown -R user:user /app
+
 RUN useradd -m -u 1000 user
+USER user
 
-COPY --chown=user ./requirements.txt requirements.txt
+COPY --chown=user . /app
+# COPY --chown=user ./requirements.txt requirements.txt
 # COPY ./requirements.txt requirements.txt
-RUN pip install -r requirements.txt
-RUN pip install imageio[ffmpeg]
 
 ENV PATH="/home/user/.local/bin:$PATH"
 
-COPY --chown=user ./run_train_infer.sh run_train_infer.sh
+# COPY --chown=user ./run_train_infer.sh run_train_infer.sh
 # COPY ./run_train_infer.sh run_train_infer.sh
 RUN chmod +x run_train_infer.sh
 
-RUN chown -R user:user /app
-USER user
 
 # Default command to run when the container starts
-CMD ["/bin/bash", "run_train_infer.sh"]
+# CMD ["/bin/bash", "run_train_infer.sh"]
+
+# Copy the current directory contents into the container at $HOME/app setting the owner to the user
+
+CMD ["python", "app.py"]