FROM nvcr.io/nvidia/cosmos/cosmos-predict2-container:1.2

WORKDIR /app

# Install additional dependencies for the API server
RUN pip install --no-cache-dir fastapi uvicorn python-multipart

# Copy handler
COPY handler.py /app/handler.py

# Expose port
EXPOSE 80

# Run the server
CMD ["uvicorn", "handler:app", "--host", "0.0.0.0", "--port", "80"]