lillybak commited on
Commit
b79351e
1 Parent(s): 77f966f

set top_n, top_k tp 1

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -80,7 +80,7 @@ recursive_index_faiss = VectorStoreIndex(nodes=base_nodes+objects, storage_conte
80
  # 1. Initalize our reranker using `FlagEmbeddingReranker` powered by the `BAAI/bge-reranker-large`.
81
  # 2. Set up our recursive query engine!
82
  reranker = FlagEmbeddingReranker(
83
- top_n=5,
84
  model="BAAI/bge-reranker-large",
85
  )
86
 
@@ -96,7 +96,7 @@ user_template = """ Think through your response step by step."""
96
  #user_query = "Who are the E-VP, Operations - and how old are they?"
97
  def resursive_fn(reranker):
98
  recursive_query_engine = recursive_index_faiss.as_query_engine(
99
- similarity_top_k=5,
100
  node_postprocessors=[reranker],
101
  verbose=True
102
  )
 
80
  # 1. Initalize our reranker using `FlagEmbeddingReranker` powered by the `BAAI/bge-reranker-large`.
81
  # 2. Set up our recursive query engine!
82
  reranker = FlagEmbeddingReranker(
83
+ top_n=1,
84
  model="BAAI/bge-reranker-large",
85
  )
86
 
 
96
  #user_query = "Who are the E-VP, Operations - and how old are they?"
97
  def resursive_fn(reranker):
98
  recursive_query_engine = recursive_index_faiss.as_query_engine(
99
+ similarity_top_k=1,
100
  node_postprocessors=[reranker],
101
  verbose=True
102
  )