File size: 523 Bytes
81cf53b
 
 
 
b7b86e2
 
 
81cf53b
 
 
 
 
 
 
 
 
b7b86e2
2cd4391
b7b86e2
 
 
 
81cf53b
771e453
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
FROM python:3.9

WORKDIR /code

# Install huggingface-cli
RUN pip install --no-cache-dir huggingface_hub

ENV REPO=TheBloke/Llama-2-7B-Chat-GGUF
ENV MODEL_NAME=llama-2-7b-chat.Q5_K_M.gguf

COPY ./requirements.txt /code/requirements.txt

RUN pip install --no-cache-dir --upgrade -r /code/requirements.txt

COPY . .

# Download the model
RUN huggingface-cli download \
    ${REPO} \
    ${MODEL_NAME} \
    --local-dir . \
    --local-dir-use-symlinks False

CMD ["uvicorn", "app:app", "--host", "0.0.0.0", "--port", "7860"]