Update app.py

app.py

@@ -9,12 +9,17 @@ import torchvision.models as models
 MODEL_PATH = "resnet50-corrosion-classifier-v1.pth"
 
 # Mappa etichette (sostituisci con le tue classi)
-IDX2LABEL = {
-    0: "Uniform corrosion",
-    1: "Pitting corrosion",
-    2: "Galvanic corrosion",
-    3: "Underfilm blistering",
-    4: "Coating delamination"
+
+    IDX2LABEL = {
+    0: "crevice_corrosion",
+    1: "erosion_corrosion",
+    2: "galvanic_corrosion",
+    3: "mic_corrosion",
+    4: "no_corrosion",
+    5: "pitting_corrosion",
+    6: "stress_corrosion",
+    7: "under_insulation_corrosion",
+    8: "uniform_corrosion"
 }
 
 device = "cuda" if torch.cuda.is_available() else "cpu"