Update Dockerfile

Dockerfile

@@ -28,6 +28,20 @@ RUN apt-get update && apt-get install -y \
 RUN mkdir -p /tmp/uploads /tmp/results /tmp/huggingface/transformers /tmp/huggingface/datasets /tmp/.u2net \
     && chmod -R 777 /tmp/uploads /tmp/results /tmp/huggingface /tmp/.u2net
 
+# Install huggingface_hub early for model downloads
+RUN pip install --no-cache-dir huggingface_hub==0.23.0
+
+# Pre-download models with error handling
+RUN if [ -z "$HF_TOKEN" ]; then \
+        echo "ERROR: HF_TOKEN is not set. Model pre-download will fail."; \
+        exit 1; \
+    else \
+        python -c "from huggingface_hub import login, snapshot_download; login(token='$HF_TOKEN'); \
+                   snapshot_download(repo_id='Intel/dpt-large', cache_dir='/tmp/huggingface', token='$HF_TOKEN')" && \
+        python -c "from huggingface_hub import snapshot_download; \
+                   snapshot_download(repo_id='depth-anything/Depth-Anything-V2-Tiny-hf', cache_dir='/tmp/huggingface', token='$HF_TOKEN')"; \
+    fi
+
 # Copy requirements file and install dependencies
 COPY requirements.txt .
 RUN pip install --no-cache-dir -U pip && \