Spaces:
Sleeping
Sleeping
Update app.py
Browse files
app.py
CHANGED
|
@@ -105,20 +105,20 @@ def recreate_query_engine():
|
|
| 105 |
# Create BM25 retriever
|
| 106 |
bm25_retriever = BM25Retriever.from_defaults(
|
| 107 |
docstore=vector_index.docstore,
|
| 108 |
-
similarity_top_k=
|
| 109 |
)
|
| 110 |
|
| 111 |
# Create vector retriever
|
| 112 |
vector_retriever = VectorIndexRetriever(
|
| 113 |
index=vector_index,
|
| 114 |
-
similarity_top_k=
|
| 115 |
similarity_cutoff=0.5
|
| 116 |
)
|
| 117 |
|
| 118 |
# Create hybrid retriever
|
| 119 |
hybrid_retriever = QueryFusionRetriever(
|
| 120 |
[vector_retriever, bm25_retriever],
|
| 121 |
-
similarity_top_k=
|
| 122 |
num_queries=1
|
| 123 |
)
|
| 124 |
|
|
|
|
| 105 |
# Create BM25 retriever
|
| 106 |
bm25_retriever = BM25Retriever.from_defaults(
|
| 107 |
docstore=vector_index.docstore,
|
| 108 |
+
similarity_top_k=15
|
| 109 |
)
|
| 110 |
|
| 111 |
# Create vector retriever
|
| 112 |
vector_retriever = VectorIndexRetriever(
|
| 113 |
index=vector_index,
|
| 114 |
+
similarity_top_k=20,
|
| 115 |
similarity_cutoff=0.5
|
| 116 |
)
|
| 117 |
|
| 118 |
# Create hybrid retriever
|
| 119 |
hybrid_retriever = QueryFusionRetriever(
|
| 120 |
[vector_retriever, bm25_retriever],
|
| 121 |
+
similarity_top_k=30,
|
| 122 |
num_queries=1
|
| 123 |
)
|
| 124 |
|