Update app.py

app.py

@@ -10,6 +10,7 @@ from langchain.chains.question_answering import load_qa_chain
 from langchain.prompts import PromptTemplate
 from dotenv import load_dotenv
 from google.api_core.exceptions import GoogleAPIError, InvalidArgument
+from google.cloud import aiplatform
 
 # Load environment variables
 load_dotenv()
@@ -34,7 +35,8 @@ def get_text_chunks(text):
 # Function to get embeddings for each chunk and save to vector store
 def get_vector_store(chunks):
     try:
-        embeddings = GoogleGenerativeAIEmbeddings(model="models/embedding-001")  
+        # embeddings = GoogleGenerativeAIEmbeddings(model="models/embedding-001")  
+        embeddings = aiplatform.GoogleGenerativeAIEmbeddings(model="text-embedding-004")
         vector_store = FAISS.from_texts(chunks, embedding=embeddings)
         vector_store.save_local("faiss_index")
     except (GoogleAPIError, InvalidArgument):