Update app.py

app.py

@@ -87,15 +87,15 @@ codes = """001 - Pendaftaran Kendaraan
 008 - Pelaporan dan Bantuan Terkait Kendaraan
 009 - Penangguhan atau Deklarasi Perubahan Penggunaan Kendaraan""".split("\n")
 
-documents = """Vehicle Registration
-Vehicle Data Update
-Vehicle Ownership or Change of Name
-Reporting of Lost Documents or Plates
-Vehicle Tax Payment and Management
-Vehicle Inspection and Verification
-Registration of Export, Import, or Fleet Vehicles
-Reporting and Assistence of Vehicles Insurance
-Suspension or Declaration of Change of Vehicle Use""".split("\n")
+# documents = """Vehicle Registration
+# Vehicle Data Update
+# Vehicle Ownership or Change of Name
+# Reporting of Lost Documents or Plates
+# Vehicle Tax Payment and Management
+# Vehicle Inspection and Verification
+# Registration of Export, Import, or Fleet Vehicles
+# Reporting and Assistence of Vehicles Insurance
+# Suspension or Declaration of Change of Vehicle Use""".split("\n")
 
 
 # examples = [
@@ -327,7 +327,7 @@ detail_perhitungan = {
         # Example formula: 5% of harga_jual
     },
     "003": {
-        "name": "Alih Kepemilikan Kendaraan",
+        "name": "Alih Kepemilikan (Balik Nama) Kendaraan",
         "formula": lambda harga_jual: harga_jual * 0.1,  
         # Example formula: 10% of harga_jual
     },
@@ -531,8 +531,8 @@ def respond(
 
     if model.config._name_or_path in model_ids[-2:]:
         # documents = [v["name"] for v in detail_perhitungan.values()]
-        # sentence_pairs = [[message, v["name"]] for v in detail_perhitungan.values()]
-        sentence_pairs = [[message, doc] for doc in documents]
+        sentence_pairs = [[message, v["name"]] for v in detail_perhitungan.values()]
+        # sentence_pairs = [[message, doc] for doc in documents]
         scores = model.predict(sentence_pairs, convert_to_tensor=True)
         weights = [1,1,1,1,1,1,1,1,1]
         # scores = [x["score"] for x in model.rank(message, documents)]