Update app.py

app.py

@@ -502,13 +502,15 @@ def respond(
     scores = cos_sim(codes_emb, text_emb).mean(axis=-1)#[:,0]
 
     scores_argsort = scores.argsort(descending=True)
+    weights = [18,8,7,6,5,4,3,2,1]
 
     # if n_num == 0:
     #     std = scores.std()
     # else:
     #     std = 0
     if n_num == 0:
-        w_avg = np.average(scores[scores_argsort].numpy(), weights=range(len(scores),0,-1))
+        # w_avg = np.average(scores[scores_argsort].numpy(), weights=range(len(scores),0,-1))
+        w_avg = np.average(scores[scores_argsort].numpy(), weights=weights)
     else:
         w_avg = 9999
     #[::-1]