Update Dockerfile

Dockerfile

@@ -1,12 +1,26 @@
+# SPDX-License-IdentifierText: NVIDIA CORPORATION & AFFILIATES. All rights reserved.
 FROM nvcr.io/nvidia/pytorch:24.10-py3
 
-# Install tools
-RUN apt-get update && apt-get install -y git ffmpeg wget build-essential ninja-build && \
-    apt-get clean && rm -rf /var/lib/apt/lists/*
+# Install system tools
+RUN apt-get update && apt-get install -y git tree ffmpeg wget build-essential && rm -rf /var/lib/apt/lists/*
 
-COPY requirements.txt /requirements.txt
-COPY app.py /app/app.py
-COPY setup_runtime.sh /setup_runtime.sh
-RUN chmod +x /setup_runtime.sh
+# Set working directory
+WORKDIR /app
 
-CMD ["/bin/bash", "/setup_runtime.sh"]
+# Copy environment and app files
+COPY ./cosmos-predict1.yaml /cosmos-predict1.yaml
+COPY ./requirements.txt /requirements.txt
+COPY ./setup.sh /setup.sh
+COPY . /app
+
+# Give permission to setup script
+RUN chmod +x /setup.sh
+
+# Run setup script as root
+RUN /bin/bash /setup.sh
+
+# Expose Gradio port
+EXPOSE 7860
+
+# Default command
+CMD ["python", "app.py"]