Update Dockerfile

Dockerfile

@@ -1,53 +1,48 @@
 FROM pytorch/pytorch:2.2.2-cuda12.1-cudnn8-runtime
 
-# Set ARG and ENV for HF token (used in snapshot_download)
-ARG HF_TOKEN
-ENV HF_TOKEN=${HF_TOKEN}
-
-# DEBUG: Print HF_TOKEN (NOT RECOMMENDED FOR PUBLIC USE)
-RUN echo "HF_TOKEN is: ${HF_TOKEN}"
-
-# # 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/**', \
-#     token='${HF_TOKEN}'\
-# )"
-
-# # --------------------------
-# # 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 \
+    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"]