| FROM python:3.9 | |
| WORKDIR /code | |
| COPY ./requirements.txt /code/requirements.txt | |
| RUN pip install --no-cache-dir --upgrade -r /code/requirements.txt | |
| COPY . . | |
| # Set the environment variables for Flask | |
| ENV FLASK_APP=app.main | |
| ENV FLASK_RUN_HOST=0.0.0.0 | |
| ENV FLASK_RUN_PORT=7860 | |
| # Set the Transformers cache directory within the container | |
| ENV TRANSFORMERS_CACHE=/code/.transformers_cache | |
| # Create the directory for Transformers cache | |
| RUN mkdir -p /code/.transformers_cache | |
| CMD ["flask", "run"] | |