Update app.py

app.py

@@ -88,46 +88,55 @@ codes = """001 - Pendaftaran Kendaraan
 
 examples = [
     {"code": "001", "examples": [
+        codes[0][6:],
         "Pendaftaran kendaraan dengan plat B 1234 XYZ",
         "Daftar baru untuk plat kendaraan D 5678 ABC",
         "Registrasi kendaraan plat AB 9102 DEF"
     ]},
     {"code": "002", "examples": [
+        codes[1][6:],
         "Pembaruan data untuk kendaraan plat B 2345 GHI",
         "Update informasi kendaraan dengan plat L 6789 JKL",
         "Perubahan data pemilik plat DK 1122 MNO"
     ]},
     {"code": "003", "examples": [
+        codes[2][6:],
         "Alih kepemilikan kendaraan dengan plat F 3344 PQR",
         "Balik nama untuk kendaraan plat W 5566 STU",
         "Perubahan pemilik pada kendaraan plat AD 7788 VWX"
     ]},
     {"code": "004", "examples": [
+        codes[3][6:],
         "Laporan kehilangan dokumen untuk kendaraan plat B 9900 YZA",
         "Hilangnya plat nomor kendaraan D 1122 BCD",
         "Dokumen kendaraan dengan plat AB 3344 EFG dilaporkan hilang"
     ]},
     {"code": "005", "examples": [
+        codes[4][6:],
         "Pembayaran pajak kendaraan dengan plat B 5566 HIJ",
         "Kelola pajak tahunan kendaraan plat L 7788 KLM",
         "Pajak kendaraan dengan plat DK 9900 NOP dibayar"
     ]},
     {"code": "006", "examples": [
+        codes[5][6:],
         "Pemeriksaan fisik kendaraan dengan plat F 1122 QRS",
         "Verifikasi kendaraan plat W 3344 TUV",
         "Cek kondisi kendaraan dengan plat AD 5566 WXY"
     ]},
     {"code": "007", "examples": [
+        codes[6][6:],
         "Pendaftaran kendaraan impor dengan plat B 7788 ZAB",
         "Registrasi armada baru dengan plat D 9900 CDE",
         "Kendaraan ekspor plat AB 1122 FGH didaftarkan"
     ]},
     {"code": "008", "examples": [
+        codes[7][6:],
         "Pelaporan terkait kendaraan dengan plat B 3344 IJK",
         "Bantuan atas kendaraan plat L 5566 LMN",
         "Laporkan masalah kendaraan dengan plat DK 7788 OPQ"
     ]},
     {"code": "009", "examples": [
+        codes[8][6:],
         "Penangguhan penggunaan kendaraan dengan plat F 9900 RST",
         "Deklarasi perubahan fungsi kendaraan plat W 1122 UVW",
         "Penggunaan baru untuk kendaraan plat AD 3344 XYZ"
@@ -228,7 +237,7 @@ model_id = model_ids[-1]
 model = SentenceTransformer(model_id, trust_remote_code=True)
 
 # codes_emb = model.encode([x[6:] for x in codes])
-codes_emb = model.encode([x["examples"] for x in examples]).mean(axis=1)
+codes_emb = model.encode([x["examples"] for x in examples])#.mean(axis=1)
 
 
 def censor_middle(number, num_to_hide=4):