Spaces:
Runtime error
Runtime error
Update model_pipeline.py
Browse files- 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 =
|
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
|