Update app.py

app.py

@@ -435,6 +435,11 @@ def respond(
     scores = cos_sim(codes_emb, text_emb).mean(axis=-1)#[:,0]
 
     scores_argsort = scores.argsort(descending=True)
+
+    if n_num == 0:
+        std = scores.std()
+    else:
+        std = 0
     #[::-1]
 
     if is_multiple:
@@ -443,17 +448,22 @@ def respond(
         request_scores = []
         # request_undetected = False
         # for i,score in enumerate(scores):
-        for i in scores_argsort:
-            if len(request_scores) >= n_num:
-                break
-            if scores[i] > threshold:
-                request_details.append(codes[i][6:])
-                request_numbers.append(codes[i][:3])
-                request_scores.append(str( round(scores[i].tolist(), 3) ) )
-
-        if not request_details:
+        if scores[scores_argsort[0]] < threshold:
             request_details.append(undetected[6:])
             request_numbers.append(undetected_code)
+        else:
+            for i in scores_argsort:
+                if scores[scores_argsort[0]] <= scores[i] - std:
+                    request_details.append(codes[i][6:])
+                    request_numbers.append(codes[i][:3])
+                    request_scores.append(str( round(scores[i].tolist(), 3) ) )
+                else:
+                    if len(request_scores) >= n_num:
+                        break
+                    if scores[i] > threshold:
+                        request_details.append(codes[i][6:])
+                        request_numbers.append(codes[i][:3])
+                        request_scores.append(str( round(scores[i].tolist(), 3) ) )
             
 
         request_numbers_copy = request_numbers