Update Dockerfile

Dockerfile

@@ -48,8 +48,10 @@ RUN useradd -m -u 1000 user
 USER user
 ENV PATH="/home/user/.local/bin:$PATH"
 
+COPY --chown=user run_train_infer.sh InstantSplat/
+
 # Set the working directory
 WORKDIR /InstantSplat
 
 # Default command to run when the container starts
-CMD ["/bin/bash"]
+CMD ["/bin/bash", "run_train_infer.sh"]