makcrx commited on
Commit
7ee7401
·
1 Parent(s): 64a1cc9
Files changed (1) hide show
  1. app.py +5 -8
app.py CHANGED
@@ -20,14 +20,11 @@ def main(query):
20
  if len(result_docs) == 0:
21
  return 'Ответ не найден', 0, ''
22
 
23
- if len(result_docs) == 1:
24
- score, index = 0, 0
25
- else:
26
- sentences = [doc[0].page_content for doc in result_docs]
27
- #print('----------------------------------------------------------------')
28
- #for doc in result_docs:
29
- # print(doc[0].metadata['articleId'], ' | ', doc[0].page_content, ' | ', doc[0].metadata['answer'])
30
- score, index = reranking.search(query, sentences)
31
 
32
  return result_docs[index][0].metadata['answer'], score, result_docs[index][0].page_content
33
 
 
20
  if len(result_docs) == 0:
21
  return 'Ответ не найден', 0, ''
22
 
23
+ #if len(result_docs) == 1:
24
+ # score, index = 0, 0
25
+ #else:
26
+ sentences = [doc[0].page_content for doc in result_docs]
27
+ score, index = reranking.search(query, sentences)
 
 
 
28
 
29
  return result_docs[index][0].metadata['answer'], score, result_docs[index][0].page_content
30