Update Dockerfile

Dockerfile

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