Ritesh-hf commited on
Commit
c095293
1 Parent(s): 24b8a68

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -88,7 +88,7 @@ retriever = PineconeHybridSearchRetriever(
88
  llm = ChatOpenAI(temperature=0, model_name="gpt-4o-mini", max_tokens=512)
89
 
90
  # Initialize Reranker
91
- reranker_model = HuggingFaceCrossEncoder(model_name="jinaai/jina-reranker-v2-base-multilingual", trust_remote_code=True)
92
  compressor = CrossEncoderReranker(model=reranker_model, top_n=10)
93
 
94
  compression_retriever = ContextualCompressionRetriever(
 
88
  llm = ChatOpenAI(temperature=0, model_name="gpt-4o-mini", max_tokens=512)
89
 
90
  # Initialize Reranker
91
+ reranker_model = HuggingFaceCrossEncoder(model_name="jinaai/jina-reranker-v2-base-multilingual", model_kwargs={"trust_remote_code":True})
92
  compressor = CrossEncoderReranker(model=reranker_model, top_n=10)
93
 
94
  compression_retriever = ContextualCompressionRetriever(