Update Dockerfile

Dockerfile

@@ -1,21 +1,17 @@
-# Use NVIDIA PyTorch base (GPU). If CPU-only, use python:3.11-slim instead.
-FROM python:3.11-slim
+FROM nvcr.io/nvidia/pytorch:24.06-py3
 
-# Set working directory
-WORKDIR /
+WORKDIR /app
 
-RUN mkdir models && chmod 777 models/
-
-# Copy requirements and install dependencies
 COPY requirements.txt .
 RUN pip install --no-cache-dir -r requirements.txt
 
+# Fix: make models and cache dirs writable
+RUN mkdir -p /app/models /app/.cache && chmod -R 777 /app/models /app/.cache
+
+ENV HF_HOME=/app/.cache
 
-# Copy the server code
 COPY . .
 
-# Expose FastAPI port
 EXPOSE 8080
 
-# Default command (can be overridden)
 CMD ["python", "server.py", "--model-repo", "SmilingWolf/wd-eva02-large-tagger-v3"]