Spaces:
Runtime error
Runtime error
Update model_pipelineV2.py
Browse files- 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)),
|
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')])
|