Update utils/complete_model.py

utils/complete_model.py

@@ -409,7 +409,10 @@ def create_classifier(checkpoint_path, model_id="facebook/dinov3-vits16-pretrain
     try:
         # A. Load Checkpoint
         checkpoint = torch.load(checkpoint_path, map_location=device, weights_only=False)
-        label_cols = checkpoint.get('label_cols', ["Class_1", "Class_2"]) # Fallback
+        label_cols = checkpoint.get('label_cols', [
+            "Cardiomegaly", "Consolidation", "Edema",
+            "Atelectasis", "Pleural Effusion", "No Findings"
+        ])
         
         # B. Load Base Model (DINO)
         logger.info("🤖 Loading DINO backbone...")