Update app.py

app.py

@@ -19,7 +19,6 @@ 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):
@@ -52,6 +51,7 @@ def init_chromadb():
 
     # Select which embeddings we want to use
     #embeddings = OpenAIEmbeddings()
+    embeddings = SentenceTransformerEmbeddings(model_name="all-MiniLM-L6-v2")
 
     # Create the vectorestore to use as the index
     vectorstore = Chroma.from_documents(texts, embeddings, persist_directory=DB_DIR)
@@ -77,7 +77,8 @@ def query_chromadb():
 
     # Select which embeddings we want to use
     #embeddings = OpenAIEmbeddings()
-    
+    embeddings = SentenceTransformerEmbeddings(model_name="all-MiniLM-L6-v2")
+
     # Load Vector store from local disk
     vectorstore = Chroma(persist_directory=DB_DIR, embedding_function=embeddings)
     #vectorstore.persist()