diff --git a/privateGPT.py b/privateGPT.py index 38fab48..4c603a2 100644 --- a/privateGPT.py +++ b/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() \ No newline at end of file + main()