jonathanjordan21 commited on
Commit
ea67fe7
·
verified ·
1 Parent(s): 173b023

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
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 = [1,1]
774
  else:
775
  text_emb = model.encode(message)
776
  scores = cos_sim(codes_emb, text_emb).mean(axis=-1)#[:,0]
777
- weights = [19,8]
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)