Update app.py

app.py

@@ -87,6 +87,16 @@ 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")
+
 
 # examples = [
 #     {"code": "001", "examples": [
@@ -521,7 +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, v["name"]] for v in detail_perhitungan.values()]
+        sentence_paris = [[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)]