Update app.py

app.py

@@ -10,6 +10,7 @@ from langchain.prompts import PromptTemplate
 from langchain_community.document_loaders import PyPDFLoader
 from langchain_chroma import Chroma
 import tempfile
+from langchain_cohere import CohereEmbeddings
 
 st.set_page_config(page_title="Document Genie", layout="wide")
 
@@ -68,7 +69,8 @@ def get_conversational_chain():
     return chain
 
 def embedding(chunk,query):
-    embeddings = GoogleGenerativeAIEmbeddings(model="models/embedding-001")
+    #embeddings = GoogleGenerativeAIEmbeddings(model="models/embedding-001")
+    embeddings = CohereEmbeddings(model="embed-english-v3.0")
     db = Chroma.from_documents(chunk,embeddings)
     doc = db.similarity_search(query)
     print(doc)