# Use NVIDIA PyTorch base (GPU). If CPU-only, use python:3.11-slim instead.
FROM python:3.11-slim

# Set working directory
WORKDIR /

# Copy requirements and install dependencies
COPY requirements.txt .
RUN pip install --no-cache-dir -r requirements.txt

# 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"]