Move vector_db loading

app.py

@@ -30,19 +30,17 @@ text_splitter = CharacterTextSplitter.from_huggingface_tokenizer(bloomz_tokenize
                                                    separator="\n")
 documents = text_splitter.split_documents(data)
 
-embeddings = HuggingFaceEmbeddings()
-
+print(documents)
 
-vectordb = Chroma.from_documents(documents=documents, embedding=embeddings)
+embeddings = HuggingFaceEmbeddings()
 
 llm = HuggingFacePipeline.from_model_id(
         model_id="bigscience/bloomz-1b7",
         task="text-generation",
         model_kwargs={"temperature" : 0, "max_length" : 500})
 
+vectordb = Chroma.from_documents(documents=documents, embedding=embeddings)
 doc_retriever = vectordb.as_retriever()
-
-
 shakespeare_qa = RetrievalQA.from_chain_type(llm=llm, chain_type="stuff", retriever=doc_retriever)
 
 def query(query):