Update app.py

app.py

@@ -17,6 +17,8 @@ logging.basicConfig(level=logging.DEBUG)
 ABS_PATH = os.path.dirname(os.path.abspath(__file__))
 DB_DIR = os.path.join(ABS_PATH, "db")
 
+embeddings = SentenceTransformerEmbeddings(model_name="all-MiniLM-L6-v2")
+#embedding_function
 
 def replace_newlines_and_spaces(text):
     # Replace all newline characters with spaces
@@ -47,15 +49,17 @@ def init_chromadb():
     texts = text_splitter.split_documents(documents)
 
     # Select which embeddings we want to use
-    embeddings = OpenAIEmbeddings()
+    #embeddings = OpenAIEmbeddings()
 
     # Create the vectorestore to use as the index
     vectorstore = Chroma.from_documents(texts, embeddings, persist_directory=DB_DIR)
     vectorstore.persist()
+    
     print(vectorstore)
     #vectorstore = None
 
     db = vectorstore
+    db.get()
     print(len(db.get()["ids"]))
 
     # Print the list of source files