Update Dockerfile

Author: codelion

Dockerfile

@@ -1,7 +1,3 @@
-LABEL org.opencontainers.image.source=https://github.com/codelion/optillm
-LABEL org.opencontainers.image.description="OptiLLM full image with model serving and API routing capabilities"
-LABEL org.opencontainers.image.licenses=Apache-2.0
-
 # Build stage
 FROM python:3.12-slim AS builder
 
@@ -30,6 +26,11 @@ RUN pip install --no-cache-dir -r requirements.txt
 # Final stage
 FROM python:3.12-slim
 
+# Add labels for the final image
+LABEL org.opencontainers.image.source=https://github.com/codelion/optillm
+LABEL org.opencontainers.image.description="OptiLLM full image with model serving and API routing capabilities"
+LABEL org.opencontainers.image.licenses=Apache-2.0
+
 # Install curl for the healthcheck
 RUN apt-get update && apt-get install -y --no-install-recommends \
   curl && \