Update model_pipelineV2.py

model_pipelineV2.py

@@ -36,7 +36,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 = 'knowledge'
         self.prompt_dir = 'prompts'
         self.child_splitter = RecursiveCharacterTextSplitter(chunk_size=200)
         self.parent_splitter = RecursiveCharacterTextSplitter(chunk_size=500)