Spaces:
Runtime error
Runtime error
File size: 503 Bytes
4b6ecd4 8ef434b c402060 e86e816 8ef434b e86e816 4b6ecd4 e86e816 c99721b c402060 c99721b 8ef434b |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 |
# 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"]
|