Upload app.py

app.py

@@ -179,7 +179,7 @@ def cal_backward(out) -> dict:
         top5_scores = -np.sort(-V)[:5]
 
     # Construction du dictionnaire pour gr.Label
-    top5_dict = {short_classes_list[int(idx)]: float(f"{score:.4f}") for idx, score in zip(top5_indices, top5_scores)}
+    top5_dict = {classes_list[int(idx)]: float(f"{score:.4f}") for idx, score in zip(top5_indices, top5_scores)}
     return top5_dict
 
 # === Chargement du modèle