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