Dockerfile

FROM python:3.9

FROM nvidia/cuda:11.4.1-cudnn8-devel-ubuntu20.04

# Install Python
RUN apt-get update && \
    apt-get install -y python3.9 python3-pip

# Install NumPy and GRPCIO
RUN pip3 install --no-cache-dir --upgrade numpy grpcio

# Set the working directory in the container
WORKDIR /code

# Copy the requirements file
COPY ./requirements.txt /code/requirements.txt

# Install dependencies
RUN pip3 install --no-cache-dir --upgrade -r /code/requirements.txt


# Copy the rest of the application code
COPY . .

# Set the default command to run your application
CMD ["python", "caption_api.py", "--host", "0.0.0.0", "--port", "7860"]