Create Dockerfile

Dockerfile

@@ -0,0 +1,18 @@
+# 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"]