Spaces:

datawithsuman
/

prompt_optimization

Paused

datawithsuman commited on Jun 25, 2024

Commit

1733ba7

verified ·

1 Parent(s): 02c3d77

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -88,7 +88,7 @@ if uploaded_files:
         # Retrieval
         bm25_retriever = BM25Retriever.from_defaults(nodes=nodes, similarity_top_k=1)
-        vector_retriever = index.as_retriever(similarity_top_k=1)
         # Hybrid Retriever class
         class HybridRetriever(BaseRetriever):

         # Retrieval
         bm25_retriever = BM25Retriever.from_defaults(nodes=nodes, similarity_top_k=1)
+        vector_retriever = index.as_retriever(similarity_top_k=3)
         # Hybrid Retriever class
         class HybridRetriever(BaseRetriever):