Spaces:
Sleeping
Sleeping
Update app.py
Browse files
app.py
CHANGED
@@ -770,11 +770,11 @@ def respond_pkb(
|
|
770 |
if type(model) == CrossEncoder:
|
771 |
sentence_pairs = [[message, v[6:]] for v in codes]
|
772 |
scores = model.predict(sentence_pairs, convert_to_tensor=True)
|
773 |
-
weights = [
|
774 |
else:
|
775 |
text_emb = model.encode(message)
|
776 |
scores = cos_sim(codes_emb, text_emb).mean(axis=-1)#[:,0]
|
777 |
-
weights = [
|
778 |
|
779 |
scores_argsort = scores.argsort(descending=True)
|
780 |
w_avg = np.average(scores[scores_argsort].numpy(), weights=weights)
|
|
|
770 |
if type(model) == CrossEncoder:
|
771 |
sentence_pairs = [[message, v[6:]] for v in codes]
|
772 |
scores = model.predict(sentence_pairs, convert_to_tensor=True)
|
773 |
+
weights = [9,8]
|
774 |
else:
|
775 |
text_emb = model.encode(message)
|
776 |
scores = cos_sim(codes_emb, text_emb).mean(axis=-1)#[:,0]
|
777 |
+
weights = [11,9]
|
778 |
|
779 |
scores_argsort = scores.argsort(descending=True)
|
780 |
w_avg = np.average(scores[scores_argsort].numpy(), weights=weights)
|