Update app.py

replace chromadb with faiss

app.py

@@ -1,9 +1,9 @@
-
 import os, tempfile, streamlit as st
 from langchain.prompts import PromptTemplate
 from langchain.chains.combine_documents import create_stuff_documents_chain
 from langchain.chains import create_retrieval_chain
-from langchain_chroma import Chroma
+# from langchain_chroma import Chroma
+from langchain.vectorstores import FAISS
 from langchain_google_genai import ChatGoogleGenerativeAI, GoogleGenerativeAIEmbeddings
 from langchain_community.document_loaders import PyPDFLoader
 
@@ -43,7 +43,7 @@ if submit:
     
                     # Generate embeddings for the pages, and store in Chroma vector database
                     embeddings = GoogleGenerativeAIEmbeddings(model="models/embedding-001")
-                    vectorstore = Chroma.from_documents(pages, embeddings)
+                    vectorstore = FAISS.from_documents(pages, embeddings)
 
                     #Configure Chroma as a retriever with top_k=5
                     st.session_state.retriever = vectorstore.as_retriever(search_kwargs={"k": 5})