Dockerfile

# Use Hugging Face's transformers image as the base image
FROM transformers:4.11.3-gpu

# Set the working directory in the container
WORKDIR /code

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

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

# Copy the application files to the container
COPY . .

# Specify the command to run on container start
CMD ["python", "caption_api.py", "--host", "0.0.0.0", "--port", "7860"]