Spaces:

datawithsuman
/

prompt_optimization

Paused

datawithsuman commited on Jun 25

Commit

a4aa40a

•

1 Parent(s): 8de4963

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -87,8 +87,8 @@ if uploaded_files:
         index = VectorStoreIndex(nodes=nodes, storage_context=storage_context)
         # Retrieval
-        bm25_retriever = BM25Retriever.from_defaults(nodes=nodes, similarity_top_k=10)
-        vector_retriever = index.as_retriever(similarity_top_k=10)
         # Hybrid Retriever class
         class HybridRetriever(BaseRetriever):

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