AlexKagan commited on
Commit
c5f5ca0
1 Parent(s): d5c894d

change generative model

Browse files
Files changed (1) hide show
  1. app.py +2 -1
app.py CHANGED
@@ -75,7 +75,8 @@ def bot(history, api_kind):
75
  # Retrieve documents relevant to query
76
  query_doc_pair = [[query, doc] for doc in documents]
77
  cross_scores = cross_encoder.predict(query_doc_pair)
78
- sim_scores_argsort = reversed(np.argsort(cross_scores))
 
79
 
80
  documents = [documents[idx] for idx in sim_scores_argsort[:top_k_rank]]
81
  logger.warning(f'num documents {len(documents)}')
 
75
  # Retrieve documents relevant to query
76
  query_doc_pair = [[query, doc] for doc in documents]
77
  cross_scores = cross_encoder.predict(query_doc_pair)
78
+ sim_scores_argsort = list(reversed(np.argsort(cross_scores)))
79
+ logger.warning(f'Finished cross encoder {len(documents)}')
80
 
81
  documents = [documents[idx] for idx in sim_scores_argsort[:top_k_rank]]
82
  logger.warning(f'num documents {len(documents)}')