Spaces:
Runtime error
Runtime error
cheesyFishes
commited on
Commit
β’
10145f4
1
Parent(s):
32e672a
fix storage path
Browse files- 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 |
-
|
|
|
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):
|