Update Dockerfile

Dockerfile

@@ -1,36 +1,41 @@
 FROM pytorch/pytorch:2.2.2-cuda12.1-cudnn8-runtime
 
-# --- Set Hugging Face environment variables ---
+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-level dependencies ---
+# Install system dependencies
 RUN apt-get update && apt-get install -y \
-    git ffmpeg curl unzip \
-    libgl1-mesa-glx \
-    libglib2.0-0 \
-    libsm6 \
-    libxrender1 \
-    libxext6 \
-    && apt-get clean
-
-# --- Set working directory ---
+    git wget curl unzip ffmpeg libgl1-mesa-glx libglib2.0-0 && \
+    apt-get clean
+
+# Set up working directory as /app
 WORKDIR /app
 
-# --- Copy all files ---
+# Copy project into /app
 COPY . /app
 
-# --- Create required cache/weights directories with permissions ---
-RUN mkdir -p /app/hf_cache /app/ckpt && chmod -R 777 /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
 
-# --- Install Python dependencies ---
-RUN pip install --upgrade pip
-RUN pip install -r requirements.txt
-RUN pip install av huggingface_hub
-RUN pip install --no-binary :all: av  # Optional: rebuild AV from source
+# List contents (for debug)
+RUN ls -la /app
 
-# --- Expose default Gradio port ---
+# Expose Gradio default port
 EXPOSE 7860
 
-# --- Start the app ---
-CMD ["python", "app.py"]
+# Start the Gradio app
+CMD ["conda", "run", "--no-capture-output", "-n", "epic", "python", "gradio_app.py"]