SwatGarg commited on
Commit
8501993
1 Parent(s): abb79b7

Update model_pipelineV2.py

Browse files
Files changed (1) hide show
  1. model_pipelineV2.py +1 -1
model_pipelineV2.py CHANGED
@@ -18,7 +18,7 @@ class ModelPipeLine:
18
  DEFAULT_DOCUMENT_PROMPT = PromptTemplate.from_template(template="{page_content}")
19
  def __init__(self):
20
  self.curr_dir = os.path.dirname(__file__)
21
- self.knowledge_dir = os.path.expanduser("~/knowledge")
22
  print("Knowledge Directory:", self.knowledge_dir)
23
  self.prompt_dir = os.path.dirname('prompts')
24
  self.child_splitter = RecursiveCharacterTextSplitter(chunk_size=200)
 
18
  DEFAULT_DOCUMENT_PROMPT = PromptTemplate.from_template(template="{page_content}")
19
  def __init__(self):
20
  self.curr_dir = os.path.dirname(__file__)
21
+ self.knowledge_dir = os.path.dirname("~/knowledge")
22
  print("Knowledge Directory:", self.knowledge_dir)
23
  self.prompt_dir = os.path.dirname('prompts')
24
  self.child_splitter = RecursiveCharacterTextSplitter(chunk_size=200)