Update app.py

app.py

@@ -400,7 +400,8 @@ def respond(
     message,
     history: list[tuple[str, str]],
     threshold,
-    is_multiple
+    is_multiple,
+    n_num,
 ):
     global codes_emb
     global undetected
@@ -429,13 +430,18 @@ def respond(
     text_emb = model.encode(message)
     scores = cos_sim(codes_emb, text_emb).mean(axis=-1)#[:,0]
 
+    scores_argsort = scores.argsort(scores)[::-1]
+
     if is_multiple:
         request_details = []
         request_numbers = []
         request_scores = []
         # request_undetected = False
-        for i,score in enumerate(scores):
-            if score > threshold:
+        # 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(score.tolist(), 3) ) )
@@ -614,7 +620,7 @@ with gr.Blocks() as demo:
         additional_inputs=[
             gr.Number(0.5, label="confidence threshold", show_label=True, minimum=0., maximum=1.0, step=0.1),
             gr.Checkbox(label="multiple", info="Allow multiple request code numbers"),
-            gr.Number(3, label="maximum number of request codes", show_label=True, step=1, precision=0)
+            gr.Number(3, label="maximum number of request codes", show_label=True, minimum=1, step=1, precision=0)
         ],
         # type="messages",
         chatbot=gr.Chatbot(height=800),