ssm123ssm commited on
Commit
177569f
1 Parent(s): ed96932

Update app.py

Browse files

Similarity k to two

Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -15,12 +15,12 @@ storage_context = StorageContext.from_defaults(persist_dir='./storage')
15
  indx = load_indices_from_storage(storage_context)
16
 
17
  from llama_index.query_engine import CitationQueryEngine
18
- query_engine = CitationQueryEngine.from_args(index=indx[1], similarity_top_k=4)
19
  list_query_engine = indx[0].as_query_engine(
20
  response_mode='tree_summarize',
21
  use_async=True,
22
  )
23
- vector_query_engine = indx[1].as_query_engine(similarity_top_k=4)
24
 
25
  import gradio as gr
26
  def chatbot(input_text, is_qen):
 
15
  indx = load_indices_from_storage(storage_context)
16
 
17
  from llama_index.query_engine import CitationQueryEngine
18
+ query_engine = CitationQueryEngine.from_args(index=indx[1], similarity_top_k=2)
19
  list_query_engine = indx[0].as_query_engine(
20
  response_mode='tree_summarize',
21
  use_async=True,
22
  )
23
+ vector_query_engine = indx[1].as_query_engine(similarity_top_k=2)
24
 
25
  import gradio as gr
26
  def chatbot(input_text, is_qen):