Update app.py

app.py

@@ -20,16 +20,18 @@ text_splitter = RecursiveCharacterTextSplitter(
     chunk_overlap = 150
 )
 
+openai.api_key = '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'
+
 # Recursive Splitting the whole text of emails into chunks
 splits = text_splitter.split_documents(documents)
 
 # Creating the Embeddings from the splits we created
-embedding = OpenAIEmbeddings(openai_api_key='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')
+embedding = OpenAIEmbeddings(openai_api_key=openai.api_key)
 
 # Storing the Embeddings into ChromaDB
 persist_directory = 'docs/chroma/'
 vectordb = Chroma.from_documents(
-    documents=splits[0:500],
+    documents=splits[0:100],
     embedding=embedding,
     persist_directory=persist_directory
 )