SwatGarg commited on
Commit
aea5285
1 Parent(s): 9435ab2

Update model_pipelineV2.py

Browse files
Files changed (1) hide show
  1. model_pipelineV2.py +1 -1
model_pipelineV2.py CHANGED
@@ -20,7 +20,7 @@ class ModelPipeLine:
20
  self.curr_dir = os.path.dirname(__file__)
21
  self.knowledge_dir = "knowledge"
22
  print("Knowledge Directory:", self.knowledge_dir)
23
- self.prompt_dir = os.path.dirname('prompts')
24
  self.child_splitter = RecursiveCharacterTextSplitter(chunk_size=200)
25
  self.parent_splitter = RecursiveCharacterTextSplitter(chunk_size=500)
26
  self.documents = process_pdf_document([os.path.join(self.knowledge_dir, 'depression_1.pdf'), os.path.join(self.knowledge_dir, 'depression_2.pdf')])
 
20
  self.curr_dir = os.path.dirname(__file__)
21
  self.knowledge_dir = "knowledge"
22
  print("Knowledge Directory:", self.knowledge_dir)
23
+ self.prompt_dir = 'prompts'
24
  self.child_splitter = RecursiveCharacterTextSplitter(chunk_size=200)
25
  self.parent_splitter = RecursiveCharacterTextSplitter(chunk_size=500)
26
  self.documents = process_pdf_document([os.path.join(self.knowledge_dir, 'depression_1.pdf'), os.path.join(self.knowledge_dir, 'depression_2.pdf')])