Update model_pipelineV2.py

model_pipelineV2.py

@@ -18,7 +18,7 @@ class ModelPipeLine:
     DEFAULT_DOCUMENT_PROMPT = PromptTemplate.from_template(template="{page_content}")
     def __init__(self):
         self.curr_dir = os.path.dirname(__file__)
-        self.knowledge_dir = os.path.join(os.path.dirname(os.path.dirname(self.curr_dir)), 'data','Knowledge_base')
+        self.knowledge_dir = os.path.join(os.path.dirname(os.path.dirname(self.curr_dir)))
         self.prompt_dir = os.path.join(os.path.dirname(os.path.dirname(self.curr_dir)), 'src','prompts')
         self.child_splitter = RecursiveCharacterTextSplitter(chunk_size=200)
         self.parent_splitter = RecursiveCharacterTextSplitter(chunk_size=500)