Update Dockerfile

Dockerfile

@@ -1,44 +1,43 @@
-FROM nvidia/cuda:12.1.1-cudnn8-runtime-ubuntu22.04
-
-ENV DEBIAN_FRONTEND=noninteractive \
-    PYTHONDONTWRITEBYTECODE=1 \
-    PYTHONUNBUFFERED=1 \
-    PIP_NO_CACHE_DIR=1 \
-    HF_HOME=/root/.cache/huggingface \
-    TRANSFORMERS_CACHE=/root/.cache/huggingface/transformers \
-    GRADIO_SERVER_NAME=0.0.0.0 \
-    GRADIO_SERVER_PORT=8080 \
-    DISPLAY=:99
-
-# System deps
-RUN apt-get update && apt-get install -y --no-install-recommends \
-    python3 python3-pip python3-venv \
-    git git-lfs \
-    ffmpeg libgl1 libgl1-mesa-dri libglib2.0-0 xvfb \
- && git lfs install \
- && rm -rf /var/lib/apt/lists/*
+FROM pytorch/pytorch:2.2.2-cuda12.1-cudnn8-runtime
 
+SHELL ["/bin/bash", "-c"]
+
+# Environment variables for Hugging Face cache
+ENV HF_HOME=/app/hf_cache
+ENV TRANSFORMERS_CACHE=/app/hf_cache
+ENV HF_TOKEN=${HF_TOKEN}
+ENV PATH=/opt/conda/bin:$PATH
+# Install system dependencies
+RUN apt-get update && apt-get install -y \
+    git wget curl unzip ffmpeg libgl1-mesa-glx libglib2.0-0 && \
+    apt-get clean
+
+# Set up working directory as /app
 WORKDIR /app
+
+# Copy project into /app
+COPY . /app
+
+# Fix permissions for all subdirectories
+RUN mkdir -p /app/pretrained /app/hf_cache /.cache/gdown && \
+    chmod -R 777 /app && \
+    chmod -R 777 /.cache && \
+    chmod -R 777 /root
+
+# Create conda environment and install dependencies
 COPY requirements.txt /app/requirements.txt
+RUN conda create -n epic python=3.10 -y && \
+    conda run -n epic pip install --upgrade pip && \
+    conda run -n epic pip install -r /app/requirements.txt
 
-# Torch first, then everything except deepspeed / flash-attn
-RUN python3 -m pip install --upgrade pip \
- && python3 -m pip install --extra-index-url https://download.pytorch.org/whl/cu121 torch torchvision torchaudio \
- && awk '!/^ *#/ && !/deepspeed/ && !/flash-attn/ && !/flash_attn/ {print $0}' requirements.txt > requirements.infer.txt \
- && python3 -m pip install -r requirements.infer.txt \
- && python3 -m pip install huggingface_hub imageio-ffmpeg
+RUN chmod -R 777 /app /workspace
 
-# App code
-COPY . /app
+# # List contents (for debug)
+RUN ls -la /app
+RUN pip install gradio
+
+# Expose Gradio default port
+EXPOSE 7860
 
-EXPOSE 8080
-ENV DEVICE=cuda \
-    SAVE_FPS=16 \
-    RESULT_DIR=/app/results \
-    MODEL_META_PATH=demo/models.json \
-    EXAMPLE_META_PATH=demo/examples.json \
-    CAMERA_POSE_META_PATH=demo/camera_poses.json \
-    DEPTH_MODEL_PATH=pretrained/Metric3D/metric_depth_vit_large_800k.pth \
-    CAPTION_MODEL_PATH=pretrained/Qwen2.5-VL-7B-Instruct
-
-CMD ["bash", "-lc", "xvfb-run -s '-screen 0 1280x720x24' python3 gradio_app.py"]
+# Start the Gradio app
+CMD ["conda", "run", "--no-capture-output", "-n", "epic", "python", "gradio_batch.py"]