Update Dockerfile

Dockerfile

@@ -1,11 +1,10 @@
 FROM pytorch/pytorch:2.2.2-cuda12.1-cudnn8-runtime
 
-# Set environment variable to control Hugging Face cache directory
+# --- Set Hugging Face environment variables ---
 ENV HF_HOME=/app/hf_cache
 ENV TRANSFORMERS_CACHE=/app/hf_cache
-ENV HF_TOKEN=${HF_TOKEN}
 
-# Install system-level dependencies
+# --- Install system-level dependencies ---
 RUN apt-get update && apt-get install -y \
     git ffmpeg curl unzip \
     libgl1-mesa-glx \
@@ -15,22 +14,23 @@ RUN apt-get update && apt-get install -y \
     libxext6 \
     && apt-get clean
 
-# Set working directory
+# --- Set working directory ---
 WORKDIR /app
 
-# Copy project files
+# --- Copy all files ---
 COPY . /app
 
-# Install Python dependencies
+# --- 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
 
-# Reinstall av from source if needed
-RUN pip install --no-binary :all: av
-
-# Expose Gradio default port
+# --- Expose default Gradio port ---
 EXPOSE 7860
 
-# Run the app
+# --- Start the app ---
 CMD ["python", "app.py"]