Update app.py

app.py

@@ -26,17 +26,20 @@ GLEASON_LABELS = {
     3: "Gleason 5"
 
 }
-BACH_LABELS = {"Benign": 0, "InSitu": 1, "Invasive": 2, "Normal": 3}
+BACH_LABELS = {0: "Benign",
+               1: "InSitu",
+               2:"Invasive",
+               3: "Normal"}
 CRC_LABELS = {
-            "ADI": 0,
-            "BACK": 1,
-            "DEB": 2,
-            "LYM": 3,
-            "MUC": 4,
-            "MUS": 5,
-            "NORM": 6,
-            "STR": 7,
-            "TUM": 8,
+            0: "ADI",
+            1: "BACK",
+            2: "DEB",
+            3: "LYM",
+            4: "MUC",
+            5: "MUS",
+            6: "NORM",
+            7: "STR",
+            8: "TUM",
         }
 
 print("Loading DinoV2 base model...")