rknl commited on
Commit
7579f1f
·
verified ·
1 Parent(s): 9b6fd33

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -3
app.py CHANGED
@@ -15,15 +15,15 @@ os.environ["OPENAI_API_KEY"] = os.getenv('oai')
15
 
16
 
17
  index = load_index_from_storage(
18
- StorageContext.from_defaults(persist_dir="./telcom_full_property_kg_processed_schema")
19
  )
20
  query_engine = index.as_query_engine(
21
  include_text=True,
22
- similarity_top_k=7,
23
  )
24
  retriever = index.as_retriever(
25
  include_text=True, # include source text, default True
26
- similarity_top_k=7,
27
  )
28
 
29
 
 
15
 
16
 
17
  index = load_index_from_storage(
18
+ StorageContext.from_defaults(persist_dir=kg_index_path)
19
  )
20
  query_engine = index.as_query_engine(
21
  include_text=True,
22
+ similarity_top_k=20,
23
  )
24
  retriever = index.as_retriever(
25
  include_text=True, # include source text, default True
26
+ similarity_top_k=20,
27
  )
28
 
29