thechaiexperiment commited on
Commit
d64968f
·
verified ·
1 Parent(s): a799501

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
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, min_query_words=2)
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)