File size: 407 Bytes
d0c7664 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 |
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"] |