SwatGarg commited on
Commit
c795b61
1 Parent(s): 724f033

Update model_pipelineV2.py

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