SwatGarg commited on
Commit
b8b4f65
1 Parent(s): 989b802

Update model_pipeline.py

Browse files
Files changed (1) hide show
  1. model_pipeline.py +1 -1
model_pipeline.py CHANGED
@@ -15,7 +15,7 @@ class ModelPipeLine:
15
  DEFAULT_DOCUMENT_PROMPT = PromptTemplate.from_template(template="{page_content}")
16
  def __init__(self):
17
  self.curr_dir = os.path.dirname(__file__)
18
- self.prompt_dir = os.path.join(os.path.dirname(os.path.dirname(self.curr_dir)), 'src','prompts')
19
  self.vectorstore, self.store = create_vectorstore()
20
  self.retriever = rag_retriever(self.vectorstore) # Create the retriever
21
  self.llm = load_llm() # Load the LLM model
 
15
  DEFAULT_DOCUMENT_PROMPT = PromptTemplate.from_template(template="{page_content}")
16
  def __init__(self):
17
  self.curr_dir = os.path.dirname(__file__)
18
+ self.prompt_dir = 'prompts'
19
  self.vectorstore, self.store = create_vectorstore()
20
  self.retriever = rag_retriever(self.vectorstore) # Create the retriever
21
  self.llm = load_llm() # Load the LLM model