Update app.py

app.py

@@ -24,15 +24,15 @@ except Exception as e:
 # Label Mapping
 # ----------------------------------
 id_to_label = {
-    0: "Dermatology",
-    1: "Neurology",
-    2: "Cardiology",
+    0: "Cardiology",
+    1: "Dermatology",
+    2: "Endocrinology",
     3: "Gastroenterology",
-    4: "Orthopedics",
-    5: "ENT",
-    6: "Pulmonology",
-    7: "Urology",
-    8: "General Medicine"
+    4: "Infectious",
+    5: "Neurology",
+    6: "Orthopedics",
+    7: "Pulmonology",
+    8: "Urology"
 }
 
 # ----------------------------------