cheesyFishes commited on
Commit
10145f4
β€’
1 Parent(s): 32e672a

fix storage path

Browse files
Files changed (1) hide show
  1. prompt_assistant.py +3 -2
prompt_assistant.py CHANGED
@@ -62,8 +62,9 @@ class Text2ImagePromptAssistant(Tool):
62
  )
63
  service_context = ServiceContext.from_defaults(llm_predictor=LLMPredictor(llm=llm))
64
  set_global_service_context(service_context)
65
-
66
- self.index = load_index_from_storage(StorageContext.from_defaults(persist_dir="./storage"))
 
67
  self.query_engine = self.index.as_query_engine(similarity_top_k=5, text_qa_template=text_qa_template, refine_template=refine_template)
68
 
69
  def __call__(self, query_str):
 
62
  )
63
  service_context = ServiceContext.from_defaults(llm_predictor=LLMPredictor(llm=llm))
64
  set_global_service_context(service_context)
65
+
66
+ storage_path = os.path.join(os.path.dirname(__file__), "storage")
67
+ self.index = load_index_from_storage(StorageContext.from_defaults(persist_dir=storage_path))
68
  self.query_engine = self.index.as_query_engine(similarity_top_k=5, text_qa_template=text_qa_template, refine_template=refine_template)
69
 
70
  def __call__(self, query_str):