Spaces:
Sleeping
Sleeping
Commit
·
55e9424
1
Parent(s):
dd922c8
Update app.py
Browse files
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)
|