| FROM python:3.10 | |
| WORKDIR /code | |
| # Install dependencies | |
| COPY ./requirements.txt /code/requirements.txt | |
| RUN pip install --no-cache-dir --upgrade -r /code/requirements.txt | |
| # Copy code | |
| COPY ./main.py /code/main.py | |
| # Set environment variables | |
| ENV AI_MODEL_ID="TheBloke/Mistral-7B-Instruct-v0.2-GPTQ" | |
| ENV PYTHONUNBUFFERED=1 | |
| # Expose port | |
| EXPOSE 7860 | |
| # Command to run the application | |
| CMD ["python", "main.py"] |