Oritsemisan commited on
Commit
5c0edc2
1 Parent(s): 2641ab6

Update model_pipelineV2.py

Browse files
Files changed (1) hide show
  1. model_pipelineV2.py +1 -1
model_pipelineV2.py CHANGED
@@ -37,7 +37,7 @@ class ModelPipeLine:
37
  def __init__(self):
38
  self.curr_dir = os.path.dirname(__file__)
39
  self.knowledge_dir = os.path.join(os.path.dirname(os.path.dirname(self.curr_dir)), 'data', 'Knowledge_base')
40
- self.prompt_dir = os.path.join(os.path.dirname(os.path.dirname(self.curr_dir)), 'src', 'prompts')
41
  self.child_splitter = RecursiveCharacterTextSplitter(chunk_size=200)
42
  self.parent_splitter = RecursiveCharacterTextSplitter(chunk_size=500)
43
  self._documents = None # Initialize as None for lazy loading
 
37
  def __init__(self):
38
  self.curr_dir = os.path.dirname(__file__)
39
  self.knowledge_dir = os.path.join(os.path.dirname(os.path.dirname(self.curr_dir)), 'data', 'Knowledge_base')
40
+ self.prompt_dir = 'prompts'
41
  self.child_splitter = RecursiveCharacterTextSplitter(chunk_size=200)
42
  self.parent_splitter = RecursiveCharacterTextSplitter(chunk_size=500)
43
  self._documents = None # Initialize as None for lazy loading