Update Dockerfile

Dockerfile

@@ -1,52 +1,50 @@
 FROM pytorch/pytorch:2.2.2-cuda12.1-cudnn8-runtime
 
+# Accept HF token as build argument and set as environment variable
 ARG HF_TOKEN
 ENV HF_TOKEN=${HF_TOKEN}
 
+# Debug print (you can remove this later)
 RUN python -c "import os; print('✅ HF_TOKEN set' if os.getenv('HF_TOKEN') else '❌ HF_TOKEN missing')"
 
-
-# # Install OS-level dependencies
-# RUN apt-get update && apt-get install -y \
-#     git ffmpeg curl \
-#     libavformat-dev libavcodec-dev libavdevice-dev libavfilter-dev \
-#     libavutil-dev libswscale-dev libswresample-dev \
-#     && apt-get clean
-
-# # Set working directory
-# WORKDIR /app
-# COPY . /app
-# RUN ls /app
-# # Create weights directory
-# # RUN mkdir -p /weights/FVSM /weights/OMSM /weights/output /weights/assets/manual_poses
-
-# # Install required Python packages
-# RUN pip install --upgrade pip
-# RUN pip install gdown
-# RUN pip install av huggingface_hub
-# RUN pip install -r requirements.txt
-# RUN pip install --no-binary :all: av  # optional av re-install
-# # --------------------------
-# # Download ckpt weights from Hugging Face dataset (requires HF_TOKEN if private)
-# RUN python -c "\
-# from huggingface_hub import snapshot_download; \
-# snapshot_download(\
-#     repo_id='roll-ai/FloVD-weights', \
-#     repo_type='dataset', \
-#     local_dir='/app/ckpt', \
-#     allow_patterns='ckpt/**', \
-#     use_auth_token=True\
-# )"
-
-# # --------------------------
-# # Download example camera pose
-# # --------------------------
-# RUN curl -L -o /weights/assets/manual_poses/example.txt https://huggingface.co/datasets/mutqa/FloVD-HF-Assets/resolve/main/example.txt
-
-# # Set permissions
-# RUN chmod -R 777 /app
-
-# EXPOSE 7860
-
-# # Start the app
-# CMD ["python", "app.py"]
+# Install OS-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 \
+    && apt-get clean
+
+# Set working directory
+WORKDIR /app
+COPY . /app
+RUN ls /app
+
+# Upgrade pip and install required Python packages
+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 "\
+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
+
+# Set permissions (optional but helpful)
+RUN chmod -R 777 /app
+
+# Expose Gradio/Streamlit/Flask port
+EXPOSE 7860
+
+# Start the app
+CMD ["python", "app.py"]