Add candidate number

app.py

@@ -5,8 +5,10 @@ model_name = "BAAI/bge-large-zh-v1.5"
 model = SentenceTransformer(model_name, device="cpu")
 
 
-def cal_sim(intent, cand1, cand2, cand3, cand4, cand5):
-    cand_list = [cand1, cand2, cand3, cand4, cand5]
+def cal_sim(*args):
+    intent = args[0]
+    cand_list = args[1:]
+    # cand_list = [cand1, cand2, cand3, cand4, cand5]
     cand_list = [cand for cand in cand_list if cand]
     embeddings_1 = model.encode([intent], normalize_embeddings=True)
     embeddings_2 = model.encode(cand_list, normalize_embeddings=True)
@@ -18,15 +20,17 @@ def cal_sim(intent, cand1, cand2, cand3, cand4, cand5):
             sim_output[i] = float(sim)
     return sim_output
 
-demo = gr.Interface(fn=cal_sim,
-                    inputs=[gr.components.Textbox(label="User query"),
-                            gr.components.Textbox(label="candidate01"),
-                            gr.components.Textbox(label="candidate02"),
-                            gr.components.Textbox(label="candidate03"),
-                            gr.components.Textbox(label="candidate04"),
-                            gr.components.Textbox(label="candidate05"),
-                            ],
-                    outputs=gr.components.Label())
-    
+
+inputs = [
+    gr.components.Textbox(label="User query"),
+]
+candidate_box = [gr.components.Textbox(label=f"candidate_{i}") for i in range(30)]
+inputs.extend(candidate_box)
+demo = gr.Interface(
+    fn=cal_sim,
+    inputs=inputs,
+    outputs=gr.components.Label(),
+)
+
 if __name__ == "__main__":
     demo.launch(share=True, debug=True)