Update app.py

app.py

@@ -73,6 +73,7 @@ from langchain.chains import RetrievalQA
 from langchain.embeddings import HuggingFaceBgeEmbeddings
 from io import BytesIO
 from langchain.document_loaders import PyPDFLoader
+from langchain.vectorstores import FAISS
 
 
 
@@ -377,7 +378,8 @@ def main():
                     chunked_documents = text_splitter.split_documents(loaded_documents)
                     persist_directory = 'db'
                     # Create and persist a Chroma vector database from the chunked documents
-                    db = Chroma.from_documents(documents=chunked_documents,embedding=embeddings,persist_directory=persist_directory)
+                    db = FAISS.from_documents(chunked_documents, embeddings)
+                    # db = Chroma.from_documents(documents=chunked_documents,embedding=embeddings,persist_directory=persist_directory)
                     db.persist()
 
                     retriever = db.as_retriever(search_kwargs={"k":1})