Upload app.py

app.py

@@ -54,6 +54,36 @@ classes_list = [
     "Serrure_Tringles",
 ]
 
+short_classes_list = [
+    "Anti-fausse-manoeuvre",
+    "Busettes",
+    "Butées",
+    "Chariots",
+    "Charnière",
+    "Compas-limiteur",
+    "Renvois d'angle",
+    "Equerres aluminium moulées",
+    "Joints à clipser",
+    "Joints à coller",
+    "Joints à glisser",
+    "Joints EPDM",
+    "Joints PVC aluminium",
+    "Silicone pour vitrage alu",
+    "Visserie inox alu",
+    "Poignée carré  7 mm",
+    "Poignée carré 8 mm",
+    "Poignée crémone",
+    "Poignée cuvette",
+    "Poignée de tirage",
+    "Poignée pour levant coulissant",
+    "Serrure crémone multipoints",
+    "Serrure cuvette",
+    "Serrure gaches",
+    "Serrure pene crochet",
+    "Serrure pour porte",
+    "Serrure tringles",
+]
+
 data_size = 384
 fpn_size = 1536
 num_classes = 27
@@ -149,7 +179,7 @@ def cal_backward(out) -> dict:
         top5_scores = -np.sort(-V)[:5]
 
     # Construction du dictionnaire pour gr.Label
-    top5_dict = {classes_list[int(idx)]: float(f"{score:.4f}") for idx, score in zip(top5_indices, top5_scores)}
+    top5_dict = {short_classes_list[int(idx)]: float(f"{score:.4f}") for idx, score in zip(top5_indices, top5_scores)}
     return top5_dict
 
 # === Chargement du modèle