Spaces:
Sleeping
Sleeping
Update app.py
Browse files
app.py
CHANGED
@@ -532,7 +532,7 @@ async def chat_endpoint(chat_query: ChatQuery):
|
|
532 |
scores = cross_encoder.predict([(query_text, doc) for doc in document_texts])
|
533 |
scored_documents = list(zip(scores, document_ids, document_texts))
|
534 |
scored_documents.sort(key=lambda x: x[0], reverse=True)
|
535 |
-
relevant_portions = extract_relevant_portions(document_texts, query_text, max_portions=3,
|
536 |
#flattened_relevant_portions = []
|
537 |
#for doc_id, portions in relevant_portions.items():
|
538 |
#flattened_relevant_portions.extend(portions)
|
|
|
532 |
scores = cross_encoder.predict([(query_text, doc) for doc in document_texts])
|
533 |
scored_documents = list(zip(scores, document_ids, document_texts))
|
534 |
scored_documents.sort(key=lambda x: x[0], reverse=True)
|
535 |
+
relevant_portions = extract_relevant_portions(document_texts, query_text, max_portions=3, chunk_size=500)
|
536 |
#flattened_relevant_portions = []
|
537 |
#for doc_id, portions in relevant_portions.items():
|
538 |
#flattened_relevant_portions.extend(portions)
|