| # Get the image of python | |
| FROM python:3.9 | |
| # Copy all the files | |
| COPY . . | |
| # Set the work directory | |
| WORKDIR / | |
| # Create a new cache directory | |
| RUN mkdir -p /path/custom_cache && chmod 777 /path/custom_cache | |
| # Set the environment's home path to the new cache file | |
| ENV TRANSFORMERS_CACHE=/path/custom_cache | |
| # Install the requirements | |
| RUN pip install --no-cache-dir -r /requirements.txt | |
| COPY . . | |
| CMD ["uvicorn", "main:app", "--host", "0.0.0.0", "--port", "7860"] | |