load .env

privateGPT.py

@@ -1,3 +1,4 @@
+from dotenv import load_dotenv
 from langchain.chains import RetrievalQA
 from langchain.embeddings import LlamaCppEmbeddings
 from langchain.callbacks.streaming_stdout import StreamingStdOutCallbackHandler
@@ -5,6 +6,8 @@ from langchain.vectorstores import Chroma
 from langchain.llms import GPT4All, LlamaCpp
 import os
 
+load_dotenv()
+
 llama_embeddings_model = os.environ.get("LLAMA_EMBEDDINGS_MODEL")
 persist_directory = os.environ.get('PERSIST_DIRECTORY')
 
@@ -51,4 +54,4 @@ def main():
             print(document.page_content)
 
 if __name__ == "__main__":
-    main()
+    main()