Update Dockerfile

Dockerfile

@@ -1,36 +1,36 @@
 FROM pytorch/pytorch:2.2.2-cuda12.1-cudnn8-runtime
 
-# Install OS-level dependencies
+# Set environment variable to control Hugging Face cache directory
+ENV HF_HOME=/app/hf_cache
+ENV TRANSFORMERS_CACHE=/app/hf_cache
+ENV HF_TOKEN=${HF_TOKEN}
+
+# Install system-level dependencies
 RUN apt-get update && apt-get install -y \
     git ffmpeg curl unzip \
-    libavformat-dev libavcodec-dev libavdevice-dev libavfilter-dev \
-    libavutil-dev libswscale-dev libswresample-dev \
+    libgl1-mesa-glx \
+    libglib2.0-0 \
+    libsm6 \
+    libxrender1 \
+    libxext6 \
     && apt-get clean
 
 # Set working directory
 WORKDIR /app
+
+# Copy project files
 COPY . /app
-RUN ls /app
 
-# Upgrade pip and install required Python packages
+# Install Python dependencies
 RUN pip install --upgrade pip
-RUN pip install gdown av huggingface_hub
 RUN pip install -r requirements.txt
-RUN pip install --no-binary :all: av  # optional: rebuild av from source
-
-# # Download ckpt weights from Hugging Face (private or public)
-# RUN python -c "import os; from huggingface_hub import snapshot_download; snapshot_download(    repo_id='roll-ai/FloVD-weights',     repo_type='dataset',     local_dir='/app/ckpt',     allow_patterns='ckpt/**',     token=os.getenv('HF_TOKEN'))"
-
-
-# # Download example camera pose
-# RUN mkdir -p /weights/assets/manual_poses
-# RUN curl -L -o /weights/assets/manual_poses/example.txt https://huggingface.co/datasets/mutqa/FloVD-HF-Assets/resolve/main/example.txt
+RUN pip install av huggingface_hub
 
-# Set permissions (optional but helpful)
-RUN chmod -R 777 /app
+# Reinstall av from source if needed
+RUN pip install --no-binary :all: av
 
-# Expose Gradio/Streamlit/Flask port
+# Expose Gradio default port
 EXPOSE 7860
 
-# Start the app
+# Run the app
 CMD ["python", "app.py"]