diff --git a/chatgpt.py b/chatgpt.py index 6ff3100a2..ff0ba42a6 100644 --- a/chatgpt.py +++ b/chatgpt.py @@ -25,6 +25,7 @@ vectorstore = Chroma(persist_directory="persist", embedding_function=OpenAIEmbeddings()) index = VectorStoreIndexWrapper(vectorstore=vectorstore) else: + #loader = TextLoader("data/data.txt") # Use this line if you only need data.txt loader = DirectoryLoader("data/") if PERSIST: index = VectorstoreIndexCreator(vectorstore_kwargs={"persist_directory":"persist"}).from_loaders([loader])