Update app.py

app.py

@@ -76,12 +76,21 @@ codes = """001 - Pendaftaran Kendaraan (Baru)
 
 undetected = "099 - Lainnya/Tidak Terdeteksi"
 
+model_ids = [
+    "sentence-transformers/paraphrase-multilingual-MiniLM-L12-v2",
+    "Alibaba-NLP/gte-multilingual-base",
+    "BAAI/bge-m3",
+    "sentence-transformers/paraphrase-multilingual-mpnet-base-v2",
+    "intfloat/multilingual-e5-small",
+    "sentence-transformers/distiluse-base-multilingual-cased-v2"
+]
 # model_id = "sentence-transformers/paraphrase-multilingual-MiniLM-L12-v2"
 # model_id = "Alibaba-NLP/gte-multilingual-base"
-model_id = "BAAI/bge-m3"
-model_id = "sentence-transformers/paraphrase-multilingual-mpnet-base-v2"
-model_id = "intfloat/multilingual-e5-small"
+# model_id = "BAAI/bge-m3"
+# model_id = "sentence-transformers/paraphrase-multilingual-mpnet-base-v2"
+# model_id = "intfloat/multilingual-e5-small"
 # model_id = "sentence-transformers/distiluse-base-multilingual-cased-v2"
+model_id = model_ids[-1]
 model = SentenceTransformer(model_id, trust_remote_code=True)
 
 codes_emb = model.encode([x[6:] for x in codes])
@@ -90,10 +99,17 @@ def respond(
     message,
     history: list[tuple[str, str]],
     threshold,
-    is_multiple
+    is_multiple,
+    chosen_model_id
 ):
     global codes_emb
     global undetected
+    global model
+    global model_id
+
+    if chosen_model_id != model_id:
+        model = SentenceTransformer(chosen_model_id, trust_remote_code=True)
+        model_id = chosen_model_id
 
     undetected_code = undetected[:3]
 
@@ -193,6 +209,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"),
+            gradio.Dropdown(choices=model_ids, value=model_ids[0])
         ]
     )