Spaces:
Runtime error
Runtime error
Prashant Kumar
commited on
Commit
•
20643b2
1
Parent(s):
89d5339
fixes spelling mistakes
Browse files- Dockerfile +1 -1
- ingest.py +1 -1
Dockerfile
CHANGED
@@ -2,7 +2,7 @@ FROM python:3.9
|
|
2 |
|
3 |
WORKDIR /app
|
4 |
|
5 |
-
COPY ./
|
6 |
RUN pip install --no-cache-dir --upgrade -r requirements.txt
|
7 |
|
8 |
COPY . /app/
|
|
|
2 |
|
3 |
WORKDIR /app
|
4 |
|
5 |
+
COPY ./requirements.txt /app/
|
6 |
RUN pip install --no-cache-dir --upgrade -r requirements.txt
|
7 |
|
8 |
COPY . /app/
|
ingest.py
CHANGED
@@ -20,7 +20,7 @@ def create_vector_db():
|
|
20 |
text_splitter = RecursiveCharacterTextSplitter(chunk_size = 500, chunk_overlap = 50)
|
21 |
texts = text_splitter.split_documents(documents)
|
22 |
|
23 |
-
embeddings = HuggingFaceEmbeddings(model_name = 'sentence-transformers/
|
24 |
db = FAISS.from_documents(texts, embeddings)
|
25 |
db.save_local(DB_FAISS_PATH)
|
26 |
|
|
|
20 |
text_splitter = RecursiveCharacterTextSplitter(chunk_size = 500, chunk_overlap = 50)
|
21 |
texts = text_splitter.split_documents(documents)
|
22 |
|
23 |
+
embeddings = HuggingFaceEmbeddings(model_name = 'sentence-transformers/all-MiniLM-L6-v2', model_kwargs = {'device': 'cpu'})
|
24 |
db = FAISS.from_documents(texts, embeddings)
|
25 |
db.save_local(DB_FAISS_PATH)
|
26 |
|