Hugging Face's logo

Spaces:
Ritesh-hf
/
adib-bank
Running

App Files Community
Ritesh-hf commited on
Commit
112720a
•
1 Parent(s): d22a473

reduce topk

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -87,7 +87,7 @@ llm = ChatPerplexity(temperature=0, pplx_api_key=GROQ_API_KEY, model="llama-3.1-
87
 
88
  # Initialize Reranker
89
  model = HuggingFaceCrossEncoder(model_name="BAAI/bge-reranker-base")
90
- compressor = CrossEncoderReranker(model=model, top_n=20)
91
 
92
  compression_retriever = ContextualCompressionRetriever(
93
  base_compressor=compressor, base_retriever=retriever
 
87
 
88
  # Initialize Reranker
89
  model = HuggingFaceCrossEncoder(model_name="BAAI/bge-reranker-base")
90
+ compressor = CrossEncoderReranker(model=model, top_n=10)
91
 
92
  compression_retriever = ContextualCompressionRetriever(
93
  base_compressor=compressor, base_retriever=retriever