thechaiexperiment commited on
Commit
55e9424
·
1 Parent(s): dd922c8

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -0
app.py CHANGED
@@ -442,6 +442,7 @@ document_ids = [doc_id for doc_id, _ in initial_results]
442
  document_texts = retrieve_document_texts(document_ids, folder_path)
443
 
444
  # Rerank the results using the CrossEncoder
 
445
  scores = cross_encoder.predict([(query_text, doc) for doc in document_texts])
446
  scored_documents = list(zip(scores, document_ids, document_texts))
447
  scored_documents.sort(key=lambda x: x[0], reverse=True)
 
442
  document_texts = retrieve_document_texts(document_ids, folder_path)
443
 
444
  # Rerank the results using the CrossEncoder
445
+ cross_encoder = models['cross_encoder']
446
  scores = cross_encoder.predict([(query_text, doc) for doc in document_texts])
447
  scored_documents = list(zip(scores, document_ids, document_texts))
448
  scored_documents.sort(key=lambda x: x[0], reverse=True)