LLM/DocsGPT
1
0
Fork 0
mirror of https://github.com/arc53/DocsGPT.git synced 2026-03-04 04:44:04 +00:00
Code Issues Packages Projects Releases Wiki Activity

Add MPNet model and update vector store for Hugging Face embeddings

Browse Source
This commit is contained in:
Alex
2024-01-09 11:39:32 +00:00
parent 001c450abb
commit b3461b7134
2 changed files with 11 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
application/Dockerfile
View File

@@ -7,6 +7,10 @@ ENV PATH="/root/.cargo/bin:${PATH}"
RUN pip install --upgrade pip && pip install tiktoken==0.5.2
COPY requirements.txt .
RUN pip install -r requirements.txt
RUN apt-get install -y wget unzip
RUN wget https://docsgpt.s3.eu-west-1.amazonaws.com/models/embeddings/mpnet-base-v2.zip
RUN unzip mpnet-base-v2.zip -d model
RUN rm mpnet-base-v2.zip
FROM python:3.11-slim-bullseye
@@ -14,6 +18,8 @@ FROM python:3.11-slim-bullseye
COPY --from=builder /usr/local/ /usr/local/
WORKDIR /app
COPY --from=builder /model /app/model
COPY . /app/application
ENV FLASK_APP=app.py
ENV FLASK_DEBUG=true