Prashant Kumar commited on
Commit
20643b2
1 Parent(s): 89d5339

fixes spelling mistakes

Browse files
Files changed (2) hide show
  1. Dockerfile +1 -1
  2. ingest.py +1 -1
Dockerfile CHANGED
@@ -2,7 +2,7 @@ FROM python:3.9
2
 
3
  WORKDIR /app
4
 
5
- COPY ./requirement.txt /app/
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/allMiniLM-L6-v2', model_kwargs = {'device': 'cpu'})
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