Update app.py

app.py

@@ -43,7 +43,18 @@ model = _load_model()
 # ---------- Features & Labels ----------
 # If present, these files define exact column order and class names
 feature_order: List[str] = json.load(open(FEATURES_PATH)) if os.path.exists(FEATURES_PATH) else []
-labels: List[str] = json.load(open(LABELS_PATH)) if os.path.exists(LABELS_PATH) else []
+
+if os.path.exists(LABELS_PATH):
+    raw_map = json.load(open(LABELS_PATH))
+    if isinstance(raw_map, dict):
+        # invert mapping → sort by value → get list of keys
+        labels: List[str] = [k.capitalize() for k, v in sorted(raw_map.items(), key=lambda x: x[1])]
+    elif isinstance(raw_map, list):
+        labels: List[str] = raw_map
+    else:
+        labels: List[str] = []
+else:
+    labels: List[str] = ["Low", "Medium", "High"]  # fallback
 
 # Fallback feature order (edit if needed; should match training names exactly)
 fallback_features = [