Spaces:
Sleeping
Sleeping
Update model_pipelineV2.py
Browse files- 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 =
|
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')])
|