Spaces:
Runtime error
Runtime error
change generative model
Browse files
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)}')
|