File size: 348 Bytes
01565f9
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
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"]