Update app.py

app.py

@@ -117,12 +117,8 @@ embedding_model = SentenceTransformer("sentence-transformers/all-MiniLM-L6-v2")
 texts = [doc.page_content for doc in docs]
 embeddings = embedding_model.encode(texts).tolist()  # Convert numpy arrays to lists
 
-# Create a wrapper function for the embedding function
-def embedding_function(texts):
-    return embedding_model.encode(texts).tolist()
-
-# Create a Chroma vector store with an embedding function and add documents and their embeddings
-vectorstore = Chroma(persist_directory="./db", embedding_function=embedding_function)
+# Create a Chroma vector store and add documents and their embeddings
+vectorstore = Chroma(persist_directory="./db", embedding_function=embedding_model.encode)
 vectorstore.add_texts(texts=texts, metadatas=[{"id": i} for i in range(len(texts))], embeddings=embeddings)
 vectorstore.persist()
 
@@ -187,3 +183,4 @@ demo.launch(debug=True)
 
 
 
+