Update Dockerfile

Dockerfile

@@ -1,23 +1,12 @@
-# SPDX-FileCopyrightText: Copyright (c) 2025 NVIDIA CORPORATION & AFFILIATES. All rights reserved.
-# SPDX-License-Identifier: Apache-2.0
-
 FROM nvcr.io/nvidia/pytorch:24.10-py3
 
-# Install basic tools
-RUN apt-get update && apt-get install -y git tree ffmpeg wget build-essential ninja-build && \
+# 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/*
 
-# Fix shell and CUDA link (optional for some systems)
-RUN rm /bin/sh && ln -s /bin/bash /bin/sh && ln -s /lib64/libcuda.so.1 /lib64/libcuda.so || true
-# Copy files into image
-COPY cosmos-predict1.yaml /cosmos-predict1.yaml
 COPY requirements.txt /requirements.txt
 COPY app.py /app/app.py
 COPY setup_runtime.sh /setup_runtime.sh
 RUN chmod +x /setup_runtime.sh
 
-# Lightweight install to keep Hugging Face build stable
-RUN pip install --no-cache-dir gradio
-
-# Default command runs the heavy setup at runtime
 CMD ["/bin/bash", "/setup_runtime.sh"]