Update Dockerfile

Dockerfile

@@ -1,48 +1,52 @@
 FROM pytorch/pytorch:2.2.2-cuda12.1-cudnn8-runtime
 
-
-
-# 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"]
+ARG HF_TOKEN
+ENV HF_TOKEN=${HF_TOKEN}
+
+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"]